[Mlir-commits] [mlir] [mlir][vector] Add a check to ensure input vector rank equals target shape rank (PR #149239)

Andrzej WarzyƄski llvmlistbot at llvm.org
Wed Jul 23 08:57:56 PDT 2025


================
@@ -224,6 +224,14 @@ struct UnrollTransferWritePattern
     SmallVector<int64_t> strides(targetShape->size(), 1);
     Location loc = writeOp.getLoc();
     ArrayRef<int64_t> originalSize = sourceVectorType.getShape();
+    // Bail-out if rank(source) != rank(target). The main limitation here is the
+    // fact that `ExtractStridedSlice` requires the rank for the input and
+    // output to match. If needed, we can relax this later.
----------------
banach-space wrote:

Is this already implemented for transfer_read?

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


More information about the Mlir-commits mailing list