File tree Expand file tree Collapse file tree 1 file changed +16
-16
lines changed
.github/workflows/frontier Expand file tree Collapse file tree 1 file changed +16
-16
lines changed Original file line number Diff line number Diff line change 1- #! /bin/bash
2-
3- n_ranks=12
4-
5- if [ " $job_device " == " gpu" ]; then
6- gpus=$( rocm-smi --showid | awk ' {print $1}' | grep -Eo ' [0-9]+' | uniq | tr ' \n' ' ' )
7- n_ranks=$( echo " $gpus " | wc -w) # number of GPUs on node
8- gpu_ids=$( echo " $gpus " | tr ' ' ' \n' | tr ' \n' ' ' | sed ' s/ $//' ) # GPU IDs from rocm-smi
9- device_opts=" --gpu -g $gpu_ids "
10- fi
11-
12- if [ " $job_device " == " gpu" ]; then
13- ./mfc.sh bench --mem 12 -j $n_ranks -o " $job_slug .yaml" -- -c frontier $device_opts -n $n_ranks
14- else
15- ./mfc.sh bench --mem 1 -j $( nproc) -o " $job_slug .yaml" -- -c frontier $device_opts -n $n_ranks
16- fi
1+ #! /bin/bash
2+
3+ n_ranks=12
4+
5+ if [ " $job_device " == " gpu" ]; then
6+ gpus=$( rocm-smi --showid | awk ' {print $1}' | grep -Eo ' [0-9]+' | uniq | tr ' \n' ' ' )
7+ n_ranks=$( echo " $gpus " | wc -w) # number of GPUs on node
8+ gpu_ids=$( echo " $gpus " | tr ' ' ' \n' | tr ' \n' ' ' | sed ' s/ $//' ) # GPU IDs from rocm-smi
9+ device_opts=" --gpu -g $gpu_ids "
10+ fi
11+
12+ if [ " $job_device " == " gpu" ]; then
13+ ./mfc.sh bench --mem 12 -j $n_ranks -o " $job_slug .yaml" -- -c frontier $device_opts -n $n_ranks
14+ else
15+ ./mfc.sh bench --mem 1 -j $( nproc) -o " $job_slug .yaml" -- -c frontier $device_opts -n $n_ranks
16+ fi
You can’t perform that action at this time.
0 commit comments