mlir._mlir_libs._mlir

MLIR Python Native Extension

Submodules

• mlir._mlir_libs._mlir.ir
• mlir._mlir_libs._mlir.passmanager
• mlir._mlir_libs._mlir.rewrite

Attributes

T
U
globals

Classes

OnExplicitAction	Generic enumeration.
CurrentLocAction	Generic enumeration.
_Globals

Functions

register_dialect(→ type)	Class decorator for registering a custom Dialect wrapper
register_operation(...)	Produce a class decorator for registering an Operation class as part of a dialect
register_op_adaptor(...)	Produce a class decorator for registering an OpAdaptor class for an operation.
register_type_caster(...)	Register a type caster for casting MLIR types to custom user types.
register_value_caster(...)	Register a value caster for casting MLIR values to custom user values.

Package Contents

mlir._mlir_libs._mlir.T

mlir._mlir_libs._mlir.U

class mlir._mlir_libs._mlir.OnExplicitAction

Bases: enum.Enum

Generic enumeration.

Derive from this class to define new enumerations.

USE_EXPLICIT = 0

USE_TRACEBACK = 1

class mlir._mlir_libs._mlir.CurrentLocAction

Bases: enum.Enum

Generic enumeration.

Derive from this class to define new enumerations.

FALLBACK = 0

NAMELOC_WRAP = 1

class mlir._mlir_libs._mlir._Globals

property dialect_search_modules: list[str]

append_dialect_search_prefix(module_name: str) → None

_check_dialect_module_loaded(dialect_namespace: str) → bool

_register_dialect_impl(dialect_namespace: str, dialect_class: object, *, replace: bool = False) → None

Testing hook for directly registering a dialect

_register_operation_impl(operation_name: str, operation_class: object, *, replace: bool = False) → None

Testing hook for directly registering an operation

loc_tracebacks_enabled() → bool

set_loc_tracebacks_enabled(arg: bool, /) → None

loc_tracebacks_frame_limit() → int

set_loc_tracebacks_frame_limit(arg: int | None) → None

register_traceback_file_inclusion(arg: str, /) → None

register_traceback_file_exclusion(arg: str, /) → None

traceback_action_on_explicit_loc() → OnExplicitAction

set_traceback_action_on_explicit_loc(arg: OnExplicitAction, /) → None

traceback_action_on_current_loc() → CurrentLocAction

set_traceback_action_on_current_loc(arg: CurrentLocAction, /) → None

mlir._mlir_libs._mlir.globals: _Globals = Ellipsis

mlir._mlir_libs._mlir.register_dialect(dialect_class: type) → type

Class decorator for registering a custom Dialect wrapper

mlir._mlir_libs._mlir.register_operation(dialect_class: type, *, replace: bool = False) → collections.abc.Callable[[type[T]], type[T]]

Produce a class decorator for registering an Operation class as part of a dialect

mlir._mlir_libs._mlir.register_op_adaptor(op_class: type, *, replace: bool = False) → collections.abc.Callable[[type[T]], type[T]]

Produce a class decorator for registering an OpAdaptor class for an operation.

mlir._mlir_libs._mlir.register_type_caster(typeid: ir.TypeID, *, replace: bool = False) → collections.abc.Callable[[collections.abc.Callable[[T], U]], collections.abc.Callable[[T], U]]

Register a type caster for casting MLIR types to custom user types.

mlir._mlir_libs._mlir.register_value_caster(typeid: ir.TypeID, *, replace: bool = False) → collections.abc.Callable[[collections.abc.Callable[[T], U]], collections.abc.Callable[[T], U]]

Register a value caster for casting MLIR values to custom user values.