tvm_tirx.tirx.compilation_pipeline.tirx_pipeline

tvm_tirx.tirx.compilation_pipeline.tirx_pipeline(*, prepare_only=False, target=None)

Prepare TIRx functions and optionally lower their module-wide ABI.

Dispatch tile instructions, lower layouts, flatten tensors, simplify indices and loops, and verify memory accesses.

Parameters:

• prepare_only (bool, optional) – Stop before finalize_tir_pipeline() when true. Defaults to false.

• target (tvm_tirx.target.Target or None, optional) – Target to bind before lowering; None preserves existing function targets.

Returns:

stages (tuple) – Lowering pass, host-finalization factory, and device-finalization factory.