Skip to content

Commit 80c6222

Browse files
authored
[TritonGPU] Run Gluon canonicalizer in LLVM lowering when ConSan is on (#9396)
Regular canonicalizer will rewrite layout conversions
1 parent df38505 commit 80c6222

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

third_party/nvidia/backend/compiler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -364,7 +364,7 @@ def make_llir(self, src, metadata, options, capability):
364364
if "consan" in options.instrumentation_mode:
365365
# Call ConcurrencySanitizerPass here, before allocating global scratch memory but after allocating tensor and shared
366366
passes.ttgpuir.add_concurrency_sanitizer(pm)
367-
passes.common.add_canonicalizer(pm)
367+
passes.gluon.add_canonicalizer(pm)
368368
passes.common.add_cse(pm)
369369
passes.ttgpuir.add_allocate_global_scratch_memory(pm)
370370
nvidia.passes.ttnvgpuir.add_proxy_fence_insertion(pm, capability)

0 commit comments

Comments
 (0)