tvm_tirx.tirx.compilation_pipeline

Contents

tvm_tirx.tirx.compilation_pipeline#

The TIR backend compilation pipeline.

Functions#

default_tir_pipeline(*[, target])

Return tirx_pipeline() with its standard lowering stages.

tirx_pipeline(*[, prepare_only, target])

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

finalize_tir_pipeline()

Split host/device functions, legalize low-precision types and lower the ABI.

finalize_host_passes()

Lower runtime builtins and target intrinsics in host functions.

finalize_device_passes()

Simplify device statements and lower target intrinsics.

register_default_tir_pipeline_selector(selector)

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

Alias

Canonical API

register_tir_pipeline

tvm_tirx.tirx.register_tir_pipeline

get_tir_pipeline

tvm_tirx.tirx.get_tir_pipeline

get_default_tir_pipeline

tvm_tirx.tirx.get_default_tir_pipeline