tvm_tirx.ir.transform.instrument.PassPrintingInstrument

tvm_tirx.ir.transform.instrument.PassPrintingInstrument#

class tvm_tirx.ir.transform.instrument.PassPrintingInstrument(print_before_pass_names, print_after_pass_names)#

Bases: PassPrintingInstrument, PassInstrument

Print IR before and/or after passes selected by exact pass name.

Parameters:

  • print_before_pass_names (Sequence[str]) – Pass names whose input IR should be printed.

  • print_after_pass_names (Sequence[str]) – Pass names whose output IR should be printed.

Examples

This observer leaves the IR unchanged:

observer = PassPrintingInstrument([], ["tirx.StmtSimplify"])
with tvm_tirx.ir.transform.PassContext(instruments=[observer]):
    after = tvm_tirx.tirx.transform.StmtSimplify()(mod)