From 23b8a00a143eaeed4dcf5126fa5fe1760a0e6123 Mon Sep 17 00:00:00 2001
From: "lcy.seso" <lcy.seso@gmail.com>
Date: Thu, 12 Sep 2024 05:23:07 +0000
Subject: [PATCH] fix the script to run tvm test.

---
 artifacts/run_all_tests_tvm.sh | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/artifacts/run_all_tests_tvm.sh b/artifacts/run_all_tests_tvm.sh
index b073263bb..926b44128 100755
--- a/artifacts/run_all_tests_tvm.sh
+++ b/artifacts/run_all_tests_tvm.sh
@@ -15,27 +15,27 @@ depths='1 2 4 8 16 32'
 hiddens='256 1024'
 for h in $hiddens; do
     for d in $depths; do
-        output_filename="$log_dir/tvm_stacked_lstm_h${h}_d${d}_.tsv"
+        output_filename=$log_dir
         python3 $rnn_bench_dir/lstm_tvm_tuned.py \
             --seq_len 64 \
-            --hidden_size h \
-            --depth d \
-            --output_file $output_filename
+            --hidden_size $h \
+            --depth $d \
+            2>&1 | tee $log_dir/tvm_stacked_lstm_h${h}_d${d}_.tsv
     done
 done
 seqs='32 64 128'
 hiddens='256 1024'
 for h in $hiddens; do
     for s in $seqs; do
-        output_filename="$log_dir/tvm_stacked_lstm_h${h}_s${s}_.tsv"
+        output_filename=$log_dir
         python3 $rnn_bench_dir/lstm_tvm_tuned.py \
-            --seq_len s \
-            --hidden_size h \
+            --seq_len $s \
+            --hidden_size $h \
             --depth 32 \
-            --output_file $output_filename
+            2>&1 | tee $log_dir/tvm_stacked_lstm_h${h}_d${d}_.tsv
     done
 done
-
+benchmark_dir="FractalTensor/benchmarks"
 # 5. run the back-2-back gemm benchmark
 b2b_gemm_dir="$benchmark_dir/fused_two_hgemms/baseline/tvm"