[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