[tests] fixup lenet argument

Signed-off-by: Avimitin <[email protected]>
chipsalliance · Aug 27, 2024 · 266ff8e · 266ff8e
1 parent 85619c9
commit 266ff8e
Showing 1 changed file with 23 additions and 21 deletions.
diff --git a/tests/pytorch/lenet/build.nix b/tests/pytorch/lenet/build.nix
@@ -29,27 +29,29 @@ buildBuddyE2ETest {
     buddy-opt subgraphs0.mlir -pass-pipeline \
         "builtin.module(func.func(tosa-to-linalg-named, tosa-to-arith, tosa-to-linalg, tosa-to-tensor))" \
       | buddy-opt \
-          -convert-elementwise-to-linalg \
-          -func-bufferize-dynamic-offset \
-          -arith-bufferize \
-          -func-bufferize \
-          -tensor-bufferize \
-          -linalg-bufferize \
-          -finalizing-bufferize \
-          -batchmatmul-optimize \
-          -convert-linalg-to-affine-loops \
-          -lower-affine \
-          -convert-vector-to-scf \
-          -convert-scf-to-cf \
-          -llvm-request-c-wrappers \
-          -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 \
+          --convert-elementwise-to-linalg \
+          --func-bufferize-dynamic-offset \
+          --arith-bufferize \
+          --func-bufferize \
+          --tensor-bufferize \
+          --linalg-bufferize \
+          --finalizing-bufferize \
+          --batchmatmul-optimize \
+          --convert-linalg-to-affine-loops \
+          --lower-affine \
+          --convert-vector-to-scf \
+          --convert-scf-to-cf \
+          --llvm-request-c-wrappers \
+          --lower-vector-exp \
+          --lower-rvv=rv32 \
+          --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 \
       > subgraphs0-lowered.mlir
 
     optArtifacts+=(