Skip to content
This repository was archived by the owner on Apr 28, 2023. It is now read-only.

Commit 065c935

Browse files
Update tuning command on FAIR cluster
This is needed following the changes to the build system.
1 parent 20b7778 commit 065c935

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tc/benchmarks/scripts/autotuner_parallel.sh

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,12 @@
88
#SBATCH --gres=gpu:2
99
#SBATCH --partition=priority,uninterrupted,learnfair,scavenge
1010

11-
export TUNER_THREADS=${TUNER_THREADS:=20}
11+
module load cuda/9.0
12+
. ${HOME}/anaconda/bin/activate
13+
conda activate tc_build
14+
export LD_LIBRARY_PATH=${CONDA_PREFIX}/lib:${LD_LIBRARY_PATH}
15+
16+
export TUNER_THREADS=${TUNER_THREADS:=8}
1217
export TUNER_DEVICES=${TUNER_DEVICES:="0,1"}
1318
export DEVICE_NAME=$(nvidia-smi -L | head -n 1 | cut -d'(' -f 1 | cut -d':' -f 2 | sed "s/ //g")
1419

0 commit comments

Comments
 (0)