[Mlir-commits] [mlir] [mlir][tensor] Apply `InsertSliceOfTransferWriteOpFolder` only when `transfer_write` overwrites all elements of `insert_slice` (PR #108803)

[Rajveer Singh Bharadwaj]

================
@@ -154,6 +163,18 @@ LogicalResult InsertSliceOfTransferWriteOpFolder::matchAndRewrite(
   return success();
 }
 
+bool InsertSliceOfTransferWriteOpFolder::doesTransferWriteCoverInsertSlice(
+    vector::TransferWriteOp writeOp) {
+  if (writeOp.getVectorType().hasStaticShape() &&
+      writeOp.getShapedType().hasStaticShape())
+    return llvm::equal(writeOp.getVectorType().getShape(),
+                       writeOp.getShapedType().getShape());
+
+  // Todo: ValueBoundsConstraintSet for dynamic shapes.
----------------
Rajveer100 wrote:

Sure thing.

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