tvm_tirx.ir.transform.ModulePass

tvm_tirx.ir.transform.ModulePass#

class tvm_tirx.ir.transform.ModulePass(pass_info)#

Bases: Pass

A pass whose callback receives and returns a complete IRModule.

Create instances with module_pass(), which accepts a function or a class defining transform_module. Call after = pass_object(before) to execute the callback under the current PassContext. Its callback, rather than this wrapper class, defines how the IR changes; see module_pass() for a TVMScript before/after example.

property pass_info#