baloo.weld package

Submodules

baloo.weld.lazy_result module

class baloo.weld.lazy_result.LazyArrayResult(weld_expr, weld_type)[source]
empty
max()[source]

Returns the maximum value.

Returns:
LazyScalarResult

The maximum value.

min()[source]

Returns the minimum value.

Returns:
LazyScalarResult

The minimum value.

class baloo.weld.lazy_result.LazyDoubleResult(weld_expr)[source]
class baloo.weld.lazy_result.LazyLongResult(weld_expr)[source]
class baloo.weld.lazy_result.LazyResult(weld_expr, weld_type, ndim)[source]

Wrapper class around a yet un-evaluated Weld result.

Attributes:
weld_expr : WeldObject or numpy.ndarray

Expression that needs to be evaluated.

weld_type : WeldType

Type of the output.

ndim : int

Dimensionality of the output.

evaluate(verbose=False, decode=True, passes=None, num_threads=1, apply_experimental_transforms=True)[source]

Evaluate the stored expression.

Parameters:
verbose : bool, optional

Whether to print output for each Weld compilation step.

decode : bool, optional

Whether to decode the result

passes : list, optional

Which Weld optimization passes to apply

num_threads : int, optional

On how many threads to run Weld

apply_experimental_transforms : bool

Whether to apply the experimental Weld transforms.

Returns:
numpy.ndarray

Output of the evaluated expression.

is_raw()[source]
values

The internal data representation.

Returns:
numpy.ndarray or WeldObject

The internal data representation.

class baloo.weld.lazy_result.LazyScalarResult(weld_expr, weld_type)[source]
class baloo.weld.lazy_result.LazyStructOfVecResult(weld_expr, weld_types)[source]
class baloo.weld.lazy_result.LazyStructResult(weld_expr, weld_types)[source]