Skip to content

Commit 57354f1

Browse files
committed
[intel] add tag to 'pm.run' calls after '625c8cb'
Signed-off-by: Anatoly Myachev <[email protected]>
1 parent 8e5a9c0 commit 57354f1

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

third_party/intel/backend/compiler.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)