jax.extend.lowering.JaxIrContext

class jax.extend.lowering.JaxIrContext(*args, **kwargs)

__init__(self) → None

Creates a new MLIR context.

The context is the top-level container for all MLIR objects. It owns the storage for types, attributes, locations, and other core IR objects. A context can be configured to allow or disallow unregistered dialects and can have dialects loaded on-demand.

Methods

__init__(self)

Creates a new MLIR context.

Attributes

allow_unregistered_dialects	Controls whether unregistered dialects are allowed in this context.
append_dialect_registry	Appends the contents of a dialect registry to the context.
attach_diagnostic_handler	Attaches a diagnostic handler that will receive callbacks.
current
d	Alias for dialects.
dialects	Gets a container for accessing dialects by name.
emit_error_diagnostics	Controls whether error diagnostics are emitted to diagnostic handlers.
enable_multithreading	Enables or disables multi-threading support in the context.
get_dialect_descriptor	Gets or loads a dialect by name, returning its descriptor object.
get_num_threads	Gets the number of threads in the context's thread pool.
is_registered_operation	Checks whether an operation with the given name is registered.
load_all_available_dialects	Loads all dialects available in the registry into the context.
set_thread_pool	Sets a custom thread pool for the context to use.