tvm_tirx.tirx.compilation_pipeline.register_default_tir_pipeline_selector

tvm_tirx.tirx.compilation_pipeline.register_default_tir_pipeline_selector#

tvm_tirx.tirx.compilation_pipeline.register_default_tir_pipeline_selector(selector)#

Register a callback selecting a pipeline for a module and target.

Parameters:

selector (Callable) – Callback accepting (module, target) and returning a pipeline tuple or None to try the next selector. Callbacks accepting target_to_bind pass it to the selected factory; other callbacks inspect the bound module, which is reused without binding it a second time.

Return type:

None