Skip to content

Commit

Permalink
fix the script to run tvm test.
Browse files Browse the repository at this point in the history
  • Loading branch information
lcy-seso committed Sep 12, 2024
1 parent 388fa4a commit 23b8a00
Showing 1 changed file with 9 additions and 9 deletions.
18 changes: 9 additions & 9 deletions artifacts/run_all_tests_tvm.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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"

Expand Down

0 comments on commit 23b8a00

Please sign in to comment.