tvm_tirx.tirx.get_tir_pipeline

tvm_tirx.tirx.get_tir_pipeline#

tvm_tirx.tirx.get_tir_pipeline(name=None, *, target=None, **kwargs)#

Construct a registered pipeline by name.

Parameters:

  • name (str or None, optional) – Registered pipeline name. None selects "default".

  • target (tvm_tirx.target.Target or None, optional) – Target to bind during lowering, after thread launches become explicit.

  • **kwargs – Additional keyword arguments forwarded to the registered factory.

Returns:

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