Skip to content

Commit 17ed62d

Browse files
author
Olivia W Hsu
committed
Add in scripts to run minmax benchmark
1 parent 45e5b8b commit 17ed62d

File tree

2 files changed

+35
-0
lines changed

2 files changed

+35
-0
lines changed

scripts/minimax_runner.sh

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
#!/bin/bash
2+
#SBATCH -N 1
3+
#SBATCH --mem 120000
4+
#SBATCH -p lanka-v3
5+
#SBATCH --exclusive
6+
7+
set -u
8+
9+
source /data/scratch/rohany/array-programming-benchmarks/venv/bin/activate
10+
11+
out=minmax-bench/numpy
12+
13+
mkdir -p "$out"
14+
mkdir -p "data/minmax"
15+
16+
jsonout="$out/statistics-numpy.json"
17+
18+
LANKA=ON NUMPY_JSON="$jsonout" make python-bench BENCHES="numpy/minmax.py::bench_minmax_statistics"

scripts/minimax_taco_runner.sh

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
#!/bin/bash
2+
#SBATCH -N 1
3+
#SBATCH --mem 120000
4+
#SBATCH -p lanka-v3
5+
#SBATCH --exclusive
6+
7+
set -u
8+
9+
out=minmax-bench/taco
10+
11+
mkdir -p "$out"
12+
13+
for i in {1..5..2}
14+
do
15+
csvout="$out/result-taco-minmax$i.csv"
16+
LANKA=ON MINMAX_ORDER="$i" TACO_CONCRETIZE_HACK=1 TACO_TENSOR_PATH="data/" TACO_OUT="$csvout" make -j8 taco-bench BENCHES="bench_minimax"
17+
done

0 commit comments

Comments
 (0)