Skip to content

Commit d5e1def

Browse files
author
Rohan Yadav
committed
scripts: add script to run FROSTT ufunc benchmarks individually
1 parent 7dce850 commit d5e1def

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

scripts/frostt_taco_runner.sh

+14
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
#!/bin/bash
2+
#SBATCH -N 1
3+
#SBATCH --mem 120000
4+
#SBATCH -p lanka-v3
5+
#SBATCH --exclusive
6+
7+
out=frostt-ufunc-bench/taco/
8+
9+
mkdir -p "$out"
10+
11+
for tensor in "nips" "uber-pickups" "chicago-crime" "enron" "nell-2" "vast"; do
12+
csvout="$out/result-$tensor.csv"
13+
LANKA=ON TACO_OUT="$csvout" make -j8 taco-bench BENCHES="bench_frostt_ufunc/$tensor*"
14+
done

0 commit comments

Comments
 (0)