[pytorch] fix matmul opt passes

Signed-off-by: Avimitin <[email protected]>
chipsalliance · Aug 14, 2024 · 603f2c2 · 603f2c2
1 parent 886638a
commit 603f2c2
Showing 1 changed file with 13 additions and 28 deletions.
diff --git a/tests/pytorch/matmul/config.nix b/tests/pytorch/matmul/config.nix
@@ -6,46 +6,31 @@
   buddyOptArgs = [
     [
       "--pass-pipeline"
-      "builtin.module(func.func(tosa-to-linalg-named, tosa-to-linalg, tosa-to-tensor, tosa-to-arith), empty-tensor-to-alloc-tensor, convert-elementwise-to-linalg, arith-bufferize, func.func(linalg-bufferize, tensor-bufferize), func-bufferize)"
+      "builtin.module(func.func(tosa-to-linalg-named, tosa-to-arith, tosa-to-linalg, tosa-to-tensor))"
     ]
     [
-      "--pass-pipeline"
-      "builtin.module(func.func(buffer-deallocation-simplification, convert-linalg-to-loops), eliminate-empty-tensors, func.func(llvm-request-c-wrappers))"
-    ]
-    [
-      "--arith-expand"
-      "--eliminate-empty-tensors"
-      "--empty-tensor-to-alloc-tensor"
-      "--one-shot-bufferize"
-      "--matmul-paralell-vectorization-optimize"
-      "--batchmatmul-optimize"
-      "--convert-linalg-to-affine-loops"
-      "--affine-loop-fusion"
-      "--affine-parallelize"
-      "--lower-affine"
-      "--convert-scf-to-openmp"
+      "--convert-elementwise-to-linalg"
       "--func-bufferize-dynamic-offset"
-      "--tensor-bufferize"
       "--arith-bufferize"
-      "--buffer-deallocation"
+      "--func-bufferize"
+      "--tensor-bufferize"
+      "--linalg-bufferize"
       "--finalizing-bufferize"
-      "--convert-vector-to-scf"
-      "--expand-strided-metadata"
-      "--cse"
+      "--batchmatmul-optimize"
+      "--convert-linalg-to-affine-loops"
+      "--lower-affine"
       "--lower-vector-exp"
       "--lower-rvv=rv32"
-      "--convert-vector-to-llvm"
-      "--memref-expand"
-      "--arith-expand"
-      "--convert-arith-to-llvm"
-      "--finalize-memref-to-llvm"
+      "--convert-vector-to-scf"
       "--convert-scf-to-cf"
       "--llvm-request-c-wrappers"
-      "--convert-openmp-to-llvm"
-      "--convert-arith-to-llvm"
+      "--convert-vector-to-llvm"
       "--convert-math-to-llvm"
       "--convert-math-to-libm"
+      "--convert-arith-to-llvm"
       "--convert-func-to-llvm"
+      "--expand-strided-metadata"
+      "--finalize-memref-to-llvm"
       "--reconcile-unrealized-casts"
     ]
   ];