[Mlir-commits] [mlir] [MLIR][memref] Fix normalization issue in memref.load (PR #107771)

[Kai Sasaki]

================
@@ -1146,7 +1147,88 @@ LogicalResult mlir::affine::replaceAllMemRefUsesWith(
       // is set.
       return failure();
     }
-    op->setOperand(memRefOperandPos, newMemRef);
+
+    // Check if it is a memref.load
+    auto memrefLoad = dyn_cast<memref::LoadOp>(op);
----------------
Lewuathe wrote:

How about extracting this part as separate function to keep the size of `replaceAllMemRefUsesWith` small? 

https://github.com/llvm/llvm-project/pull/107771