Skip to content

Commit 37a00ab

Browse files
authored
[UT] enable common loop fusion optimization (#3471)
**FuseNestedLoopsPass** is a common pass independent with backend, so we just enable it in our Intel backend.
1 parent 8628b79 commit 37a00ab

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

python/test/unit/language/test_core.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6708,8 +6708,6 @@ def test_tl_range_num_stages(device):
67086708

67096709

67106710
def test_tl_range_fuse():
6711-
if is_xpu():
6712-
pytest.skip("loop fusion is not enabled on XPU")
67136711
if is_hip():
67146712
pytest.skip("loop fusion is not enabled on AMD")
67156713

third_party/intel/backend/compiler.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -283,6 +283,7 @@ def make_ttgir(mod, metadata, opt, properties):
283283
intel.passes.ttgpuir.add_rewrite_tensor_pointer(pm)
284284
intel.passes.ttgpuir.add_pipeline(pm, opt.num_stages, False)
285285

286+
passes.ttgpuir.add_fuse_nested_loops(pm)
286287
passes.ttgpuir.add_optimize_thread_locality(pm)
287288
passes.ttgpuir.add_optimize_dot_operands(pm, True)
288289
passes.common.add_cse(pm)

0 commit comments

Comments
 (0)