tvm_tirx.backend.trn.pipeline

Contents

tvm_tirx.backend.trn.pipeline#

Trainium TIRX pipeline entrypoints.

Functions#

trn_pipeline(*[, target])

Construct the Trainium lowering and host/device finalization stages.

finalize_device_passes_trn()

Construct the statement-simplification stage for Trainium device IR.