[Mlir-commits] [mlir] [mlir][vector] Support warp distribution of `transfer_read` with dependencies (PR #77779)

Nicolas Vasilache llvmlistbot at llvm.org
Fri Jan 12 01:43:37 PST 2024


================
@@ -822,6 +822,10 @@ struct WarpOpTransferRead : public OpRewritePattern<WarpExecuteOnLane0Op> {
       return failure();
     auto read = operand->get().getDefiningOp<vector::TransferReadOp>();
 
+    // Source must be defined outside of the region.
+    if (!warpOp.isDefinedOutsideOfRegion(read.getSource()))
+      return failure();
----------------
nicolasvasilache wrote:

while we are here, can we return `notifyMatchFailure("Source must be defined outside of the region.")` ?
Ideally also fix the rest if possible.

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


More information about the Mlir-commits mailing list