Skip to content

Commit 4991020

Browse files
committed
Remove RewriteTensorPointer from the optimization pipeline
Signed-off-by: Tiotto, Ettore <[email protected]>
1 parent 22b7ec9 commit 4991020

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

third_party/intel/backend/compiler.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -235,7 +235,8 @@ def make_ttgir(mod, metadata, opt, properties):
235235
intel.passes.ttgpuir.add_accelerate_matmul(pm)
236236
intel.passes.ttgpuir.add_remove_layout_conversions(pm)
237237
intel.passes.ttgpuir.add_materialize_block_pointer(pm)
238-
intel.passes.ttgpuir.add_rewrite_tensor_pointer(pm)
238+
if os.getenv("TRITON_INTEL_REWRITE_TENSOR_POINTER", "0") == "1":
239+
intel.passes.ttgpuir.add_rewrite_tensor_pointer(pm)
239240
intel.passes.ttgpuir.add_pipeline(pm, opt.num_stages, False)
240241

241242
intel.passes.ttgpuir.add_coalesce(pm)

0 commit comments

Comments
 (0)