@@ -105,7 +105,7 @@ def make_ttgir(mod, metadata, opt):
105105 passes .common .add_cse (pm )
106106 intel .passes .ttgpuir .add_schedule_load (pm )
107107 passes .common .add_symbol_dce (pm )
108- pm .run (mod )
108+ pm .run (mod , 'make_ttgir_adv_path' )
109109 return mod
110110
111111 @staticmethod
@@ -200,7 +200,7 @@ def annotate_module(mod, properties, opt, target_arch):
200200 module_opts .threads_per_warp = opt .warp_size
201201 module_opts .target_arch = target_arch
202202 intel .passes .ttgpuir .add_triton_annotate_module (pm , module_opts )
203- pm .run (mod )
203+ pm .run (mod , 'annotate_module' )
204204
205205 @staticmethod
206206 def get_split_barrier_scope (opt ):
@@ -227,7 +227,7 @@ def make_ttir(mod, metadata, opt):
227227 passes .common .add_cse (pm )
228228 passes .common .add_symbol_dce (pm )
229229 passes .ttir .add_loop_unroll (pm )
230- pm .run (mod )
230+ pm .run (mod , 'make_ttir' )
231231 return mod
232232
233233 @staticmethod
@@ -287,7 +287,7 @@ def make_ttgir(mod, metadata, opt, properties):
287287 if knobs .intel .opt_reduction_locality :
288288 intel .passes .ttgpuir .add_optimize_reduction_locality (pm )
289289 intel .passes .arith .add_arith_emulate_unsupported_floats (pm , ["bf16" ], "f32" )
290- pm .run (mod )
290+ pm .run (mod , 'make_ttgir' )
291291 metadata ["cluster_dims" ] = (cluster_info .clusterDimX , cluster_info .clusterDimY , cluster_info .clusterDimZ )
292292 return mod
293293
@@ -303,7 +303,7 @@ def gluon_to_ttgir(self, src, metadata, options):
303303 passes .gluon .add_canonicalizer (pm )
304304 passes .ttgpuir .add_combine_tensor_select_and_if (pm )
305305
306- pm .run (mod )
306+ pm .run (mod , 'gluon_to_ttgir' )
307307 metadata ["tensordesc_meta" ] = mod .get_tensordesc_metadata ()
308308 return mod
309309
@@ -338,7 +338,7 @@ def make_llir(src, metadata, options):
338338 passes .llvmir .add_di_scope (pm )
339339 if XPUBackend .instrumentation :
340340 XPUBackend .instrumentation .patch ("llvmir_to_llvm" , pm , mod .context )
341- pm .run (mod )
341+ pm .run (mod , 'make_llir' )
342342 # LLVM-IR (MLIR) -> LLVM-IR (LLVM)
343343 llvm .init_targets ()
344344 context = llvm .context ()
0 commit comments