[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:36 PST 2024
================
@@ -1315,6 +1298,11 @@ struct WarpOpExtractElement : public OpRewritePattern<WarpExecuteOnLane0Op> {
unsigned int operandNumber = operand->getOperandNumber();
auto extractOp = operand->get().getDefiningOp<vector::ExtractElementOp>();
VectorType extractSrcType = extractOp.getSourceVectorType();
+ // TODO: Supported shuffle types should be parameterizable, similar to
+ // `WarpShuffleFromIdxFn`.
+ if (!extractSrcType.getElementType().isF32() &&
+ !extractSrcType.getElementType().isInteger(32))
+ return failure();
----------------
nicolasvasilache wrote:
notify everywhere plz
https://github.com/llvm/llvm-project/pull/77779
More information about the Mlir-commits
mailing list