We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 98437d2 commit e7e7de8Copy full SHA for e7e7de8
.github/workflows/frontier/bench.sh
@@ -2,14 +2,14 @@
2
3
n_ranks=12
4
5
-if [ "$job_device" == "gpu" ]; then
+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
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
0 commit comments