Skip to content

Commit 01bb1d7

Browse files
authored
[AMD] Use -triton-licm to replace general -licm (#6212)
This picks up #6051 to enable hoisting loop invariant loads in the AMD backend.
1 parent d392203 commit 01bb1d7

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

third_party/amd/backend/compiler.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,7 @@ def make_ttir(mod, metadata, options):
208208
passes.ttir.add_combine(pm)
209209
passes.ttir.add_reorder_broadcast(pm)
210210
passes.common.add_cse(pm)
211-
passes.common.add_licm(pm)
211+
passes.ttir.add_triton_licm(pm)
212212
passes.common.add_symbol_dce(pm)
213213
passes.ttir.add_loop_unroll(pm)
214214
pm.run(mod)
@@ -234,7 +234,7 @@ def make_ttgir(mod, metadata, options):
234234

235235
passes.ttgpuir.add_fuse_nested_loops(pm)
236236
passes.common.add_canonicalizer(pm)
237-
passes.common.add_licm(pm)
237+
passes.ttir.add_triton_licm(pm)
238238
passes.common.add_canonicalizer(pm)
239239

240240
global_prefetch = int(os.getenv("TRITON_HIP_GLOBAL_PREFETCH", "0"))

0 commit comments

Comments
 (0)