[Mlir-commits] [mlir] [MLIR] Allowing unsupported conv2d op to fail gracefully from vectorization (PR #130181)
Zhuoran Yin
llvmlistbot at llvm.org
Tue Mar 11 07:40:40 PDT 2025
================
@@ -3865,27 +3923,13 @@ struct Conv1DGenerator
oper = Pool;
isPoolExt = true;
poolExtOp = feedOp->getName().getIdentifier();
- } else if (!((isa<arith::MulIOp, arith::MulFOp>(feedOp) ||
- (isa<arith::AndIOp>(feedOp) &&
- feedOp->getResultTypes()[0].isInteger(1))) &&
- llvm::all_of(feedOp->getOperands(), [](Value v) {
- if (isa<BlockArgument>(v))
- return true;
- if (Operation *op = v.getDefiningOp())
- return isCastOfBlockArgument(op);
- return false;
- }))) {
- return false;
+ } else {
+ oper = Conv;
----------------
jerryyin wrote:
This does make it cleaner.
https://github.com/llvm/llvm-project/pull/130181
More information about the Mlir-commits
mailing list