Skip to content

mul

Classes

MulOperator 

MulOperator(op: AirOperator, model: AirModel, platform: SocPlatform, prefix: str = 'aot', attributes: dict[str, str] = {})

MUL operator.

This operator computes the element-wise multiplication of two tensors.

Parameters:

  • op

    (AirOperator) –

    The AIR operator to wrap.

  • model

    (AirModel) –

    The AIR model.

  • platform

    (SocPlatform) –

    The target platform for code generation.

  • prefix

    (str, default: 'aot' ) –

    Prefix for generated code files. Defaults to "aot".

  • attributes

    (dict[str, str], default: {} ) –

    Attributes for template values. Defaults to {}.

Attributes

has_init property 
has_init: bool

MUL initialization is a no-op once routed through the shared kernel.

Functions

shared_kernel_helpers 
shared_kernel_helpers() -> list[KernelHelper]

Route the run body through the shared elementwise-multiply kernel.

compute_values 
compute_values() -> dict[str, str]

Compute the values for the operator.

Returns:

  • dict[str, str] –

    dict[str, str]: Computed values for the operator.

emit 
emit(save_path: Path)

Generate the source code for the operator.

Parameters:

  • save_path
    (Path) –

    Path to save the generated code.

Functions