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

Diego Caballero llvmlistbot at llvm.org
Wed Feb 14 15:27:48 PST 2024


================
@@ -1609,6 +1610,20 @@ static LogicalResult reductionPreconditions(LinalgOp op) {
 }
 
 static LogicalResult vectorizeDynamicLinalgOpPrecondition(linalg::LinalgOp op) {
+  if (auto conv = dyn_cast<linalg::DepthwiseConv1DNwcWcOp>(op.getOperation())) {
----------------
dcaballe wrote:

could we move all this conv specific logic to a specific `*Precondition` function?

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


More information about the Mlir-commits mailing list