[Mlir-commits] [mlir] [mlir][linalg] Enable masked vectorisation for depthwise convolutions (PR #81625)

Cullen Rhodes llvmlistbot at llvm.org
Wed Mar 6 05:50:56 PST 2024


================
@@ -1608,7 +1610,31 @@ static LogicalResult reductionPreconditions(LinalgOp op) {
   return success();
 }
 
+static LogicalResult vectorizeDynamicConvOpPrecondition(linalg::LinalgOp conv) {
+  if (!isa<linalg::DepthwiseConv1DNwcWcOp>(conv.getOperation())) {
----------------
c-rhodes wrote:

think `.getOperation()` can be dropped
```suggestion
  if (!isa<linalg::DepthwiseConv1DNwcWcOp>(conv)) {
```

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


More information about the Mlir-commits mailing list