Compilation

tvm.tirx.build

tvm.tirx.build(mod: PrimFunc | IRModule, target: str | Target | None = None, pipeline: None | str | Pass = 'default')

Build a function with a signature, generating code for devices coupled with target information.

Parameters:

• mod (Union[PrimFunc, IRModule]) – The input to be built.

• target (Optional[Union[str, Target]]) – The target for compilation.

• pipeline (Union[None, str, tvm.transform.Pass]) – The pipeline to use for compilation.

Returns:

A module combining both host and device code.

Return type:

tvm.runtime.Module

tvm.tirx.compilation_pipeline

The TIR backend compilation pipeline.

tvm.tirx.compilation_pipeline.default_tir_pipeline()

The default tirx pipeline used in tvm.tirx.build

tvm.tirx.compilation_pipeline.tirx_pipeline()

The TIRX pipeline used in tvm.tirx.build

tvm.tirx.compilation_pipeline.finalize_host_passes()

The default finalization passes for TIR backend.

tvm.tirx.compilation_pipeline.finalize_device_passes()

The default finalization passes for TIR backend.

tvm.tirx.compilation_pipeline.finalize_device_passes_tirx()

The TIRx finalization passes for TIR backend.

tvm.tirx.compilation_pipeline.register_tir_pipeline(name: str, pipeline_factory) → None

Register a named TIR pipeline factory.

tvm.tirx.compilation_pipeline.get_tir_pipeline(name: str | None = None, **kwargs) → Pass

Get pre-build pipeline by name

Parameters:

name (Optional[str]) – Name of the pipeline

tvm.tirx.compilation_pipeline.get_default_tir_pipeline(target: Target) → Pass

Get the default TIR pipeline for the given target.