[OPTIMIZER] Added memory coalescing pass (#31)

2022-07-31 20:59:31 -07:00
parent d1593e6ca8
commit 3236642e8f
7 changed files with 180 additions and 4 deletions
--- a/lib/Analysis/AxisInfo.cpp
+++ b/lib/Analysis/AxisInfo.cpp
@@ -94,7 +94,8 @@ ChangeResult AxisInfoAnalysis::visitOperation(
  AxisInfo curr;
  // This preserves the input axes (e.g., cast):
  if (llvm::isa<arith::ExtSIOp, arith::ExtUIOp, arith::TruncIOp,
-                triton::PtrToIntOp, triton::IntToPtrOp>(op))
+                triton::PtrToIntOp, triton::IntToPtrOp,
+                triton::gpu::ConvertLayoutOp>(op))
    curr = operands[0]->getValue();
  // Constant ranges
  if (triton::MakeRangeOp make_range =