Skip to content

Commit 99fd4c2

Browse files
authored
Update bench.sh
1 parent 105ae27 commit 99fd4c2

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

.github/workflows/phoenix/bench.sh

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22

33
n_ranks=12
44

5+
export TMPDIR=/storage/home/hcoda1/6/sbryngelson3/scratch/runner-tmp
6+
57
if [ "$job_device" == "gpu" ]; then
68
n_ranks=$(nvidia-smi -L | wc -l) # number of GPUs on node
79
gpu_ids=$(seq -s ' ' 0 $(($n_ranks-1))) # 0,1,2,...,gpu_count-1

0 commit comments

Comments
 (0)