[Mlir-commits] [mlir] [mlir][SVE] Add e2e for 1D depthwise WC convolution (PR #85225)

Andrzej Warzyński llvmlistbot at llvm.org
Sat Mar 23 11:23:28 PDT 2024


================
@@ -0,0 +1,60 @@
+// DEFINE: %{compile} =  mlir-opt %s \
+// DEFINE:    -transform-interpreter -test-transform-dialect-erase-schedule \
+// DEFINE:    -one-shot-bufferize="bufferize-function-boundaries" -lower-vector-mask -cse -canonicalize -convert-vector-to-scf -arm-sve-legalize-vector-storage \
+// DEFINE:    -convert-vector-to-llvm="enable-arm-sve" -test-lower-to-llvm -o %t
+// DEFINE: %{entry_point} = conv
+// DEFINE: %{run} = %mcr_aarch64_cmd %t -e %{entry_point} -entry-point-result=void --march=aarch64 --mattr="+sve"\
+// DEFINE:    -shared-libs=%mlir_runner_utils,%mlir_c_runner_utils
+
+// RUN: %{compile} | %{run} | FileCheck %s
----------------
banach-space wrote:

That could be the case 🤔

There's https://github.com/llvm/llvm-project/blob/main/mlir/test/Integration/Dialect/Linalg/CPU/ArmSVE/matmul.mlir that works fine with this configuration - we probably just missed that it failed first time it was run. Also a guess :)

If I don't hear back from Omair with other suggestion, I'll merge the updated version around Monday and just monitor the bots to see what happens. Thanks Ben!

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


More information about the Mlir-commits mailing list