[Mlir-commits] [mlir] [Linalg] Update Vectorization to work with both named as well as generic conv ops (PR #176339)

Han-Chung Wang llvmlistbot at llvm.org
Fri Jan 16 14:08:23 PST 2026


================
@@ -2431,10 +2432,11 @@ static LogicalResult vectorizeLinalgOpPrecondition(
   if (isElementwise(linalgOp))
     return success();
 
-  // TODO: isaConvolutionOpInterface that can also infer from generic
-  // features. But we will still need stride/dilation attributes that will be
-  // annoying to reverse-engineer...
-  if (isa<ConvolutionOpInterface>(linalgOp.getOperation()))
+  // Check for convolution ops - both named ops implementing
+  // ConvolutionOpInterface and generic ops that semantically match convolution
+  // patterns.
+  if (isa<ConvolutionOpInterface>(linalgOp.getOperation()) ||
----------------
hanhanW wrote:

Do we still need `isa<ConvolutionOpInterface>`? I thought the other one is more robust than `isa<>` method.

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


More information about the Mlir-commits mailing list