mlir.dialects._transform_tune_extension_ops_gen

Attributes

_ods_ir
_Buffer

Classes

AlternativesOp	This op represents a choice over which of its regions is to be used.
AlternativesOpAdaptor
KnobOp	Provides a representation for "tunables" within schedules.
KnobOpAdaptor

Functions

tune_alternatives(→ Union[_ods_ir, _ods_ir, ...)
tune_knob(→ _ods_ir)

Module Contents

mlir.dialects._transform_tune_extension_ops_gen._ods_ir

mlir.dialects._transform_tune_extension_ops_gen._Buffer

class mlir.dialects._transform_tune_extension_ops_gen.AlternativesOp(results_: Sequence[_ods_ir], name: Any | _ods_ir, num_alternatives: int, *, selected_region_attr: Any | _ods_ir | None = None, selected_region_param: _ods_ir | None = None, loc: _ods_ir | None = None, ip: _ods_ir | None = None)

Bases: _ods_ir

This op represents a choice over which of its regions is to be used.

When selected_region is provided, the semantics are that this op is to be substituted for by the selected region, meaning the region’s results become the results of this op. Without a provided selected_region, the semantics are that this non-deterministic choice is yet to be resolved – which in terms of the op’s interpreted semantics is a failure.

The selected_region argument is either an IntegerAttr or a param holding an IntegerAttr, which should provide a valid zero-based index with respect to the number of alternatives, i.e. regions.

OPERATION_NAME = 'transform.tune.alternatives'

_ODS_REGIONS = (0, False)

selected_region_param() → _ods_ir | None

name() → _ods_ir

selected_region_attr() → _ods_ir | None

results_() → _ods_ir

alternatives() → _ods_ir

class mlir.dialects._transform_tune_extension_ops_gen.AlternativesOpAdaptor

Bases: _ods_ir

OPERATION_NAME = 'transform.tune.alternatives'

selected_region_param() → _ods_ir | None

name() → _ods_ir

selected_region_attr() → _ods_ir | None

mlir.dialects._transform_tune_extension_ops_gen.tune_alternatives(results_: Sequence[_ods_ir], name: Any | _ods_ir, num_alternatives: int, *, selected_region_attr: Any | _ods_ir | None = None, selected_region_param: _ods_ir | None = None, loc: _ods_ir | None = None, ip: _ods_ir | None = None) → _ods_ir | _ods_ir | AlternativesOp

class mlir.dialects._transform_tune_extension_ops_gen.KnobOp(result: _ods_ir, name: Any | _ods_ir, options: Any | _ods_ir, *, selected: Any | _ods_ir | None = None, loc: _ods_ir | None = None, ip: _ods_ir | None = None)

Bases: _ods_ir

Provides a representation for “tunables” within schedules.

Each op represents a single tunable, which has a name and a set of valid options described by an attribute. Without a specified selected option, this op represents a non-deterministic choice that has yet to be resolved – as such, the interpreter runtime semantics is to raise a failure.

The non-deterministic choice is resolved through providing a selected attribute. When provided, the interpreter runtime semantics are to return the selected attribute as a param through the op’s result.

In case the options attribute is an ArrayAttr, the verifier checks that the provided selected attribute occurs in options.

OPERATION_NAME = 'transform.tune.knob'

_ODS_REGIONS = (0, True)

name() → _ods_ir

options() → _ods_ir

selected() → _ods_ir | None

result() → _ods_ir

class mlir.dialects._transform_tune_extension_ops_gen.KnobOpAdaptor

Bases: _ods_ir

OPERATION_NAME = 'transform.tune.knob'

name() → _ods_ir

options() → _ods_ir

selected() → _ods_ir | None

mlir.dialects._transform_tune_extension_ops_gen.tune_knob(result: _ods_ir, name: Any | _ods_ir, options: Any | _ods_ir, *, selected: Any | _ods_ir | None = None, loc: _ods_ir | None = None, ip: _ods_ir | None = None) → _ods_ir