[Mlir-commits] [mlir] [mlir][linalg] Enable masked vectorisation for depthwise convolutions (PR #81625)
Andrzej WarzyĆski
llvmlistbot at llvm.org
Thu Mar 14 02:18:53 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>) {
----------------
banach-space wrote:
Actually, these shapes were meant to be dynamic, sorry and thanks for catching this! (the example won't vectorise anyway, but the point was to catch CW convs with **dynamic** shapes rather than CW convs with **static** shapes)
https://github.com/llvm/llvm-project/pull/81625
More information about the Mlir-commits
mailing list