[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