[Mlir-commits] [mlir] [mlir][linalg] Enable masked vectorisation for depthwise convolutions (PR #81625)
Cullen Rhodes
llvmlistbot at llvm.org
Tue Mar 12 06:08:32 PDT 2024
================
@@ -19,10 +19,47 @@ module attributes {transform.with_named_sequence} {
// -----
-func.func @depthwise_conv1d_nwc_wc_dyn_ch_dim(%input: memref<3x5x?xf32>, %filter: memref<2x?xf32>, %output: memref<3x2x?xf32>) {
+func.func @depthwise_conv1d_ncw_cw(%input: memref<3x5x4xf32>, %filter: memref<5x1xf32>, %output: memref<3x5x4xf32>) {
----------------
c-rhodes wrote:
please could you add a comment to say the channel dim must be dynamic to vectorize?
https://github.com/llvm/llvm-project/pull/81625
More information about the Mlir-commits
mailing list