tvm_tirx.tirx.register_tir_pipeline

tvm_tirx.tirx.register_tir_pipeline(name, pipeline_factory)

Register a factory returning lowering and host/device finalization stages.

Factories accepting an explicit target keyword own target binding. They must lower S-TIR thread bindings before binding the complete module and specializing helpers. Other factories receive inputs with targets already bound; they cannot classify calls inside unlowered S-TIR thread bindings.

Parameters:

• name (str) – Name used to select the pipeline; replaces any existing registration.

• pipeline_factory (Callable) – Factory returning the lowering pass and host/device finalization factories.

Return type:

None