Skip to content

Commit ba33af4

Browse files
authored
Attempt to fix benchmarking on Phoenix (#849)
1 parent dbd40d3 commit ba33af4

File tree

2 files changed

+60
-2
lines changed

2 files changed

+60
-2
lines changed

.github/workflows/phoenix/bench.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ if [ "$job_device" == "gpu" ]; then
99
fi
1010

1111
if ["$job_device" == "gpu"]; then
12-
./mfc.sh bench --mem 12 -j $(nproc) -o "$job_slug.yaml" -- -c phoenix $device_opts -n $n_ranks
12+
./mfc.sh bench --mem 12 -j $(nproc) -o "$job_slug.yaml" -- -c phoenix-bench $device_opts -n $n_ranks
1313
else
14-
./mfc.sh bench --mem 1 -j $(nproc) -o "$job_slug.yaml" -- -c phoenix $device_opts -n $n_ranks
14+
./mfc.sh bench --mem 1 -j $(nproc) -o "$job_slug.yaml" -- -c phoenix-bench $device_opts -n $n_ranks
1515
fi
Lines changed: 58 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,58 @@
1+
#!/usr/bin/env bash
2+
3+
<%namespace name="helpers" file="helpers.mako"/>
4+
5+
% if engine == 'batch':
6+
#SBATCH --nodes=${nodes}
7+
#SBATCH --ntasks-per-node=${tasks_per_node}
8+
#SBATCH --job-name="${name}"
9+
#SBATCH --output="${name}.out"
10+
#SBATCH --time=${walltime}
11+
% if account:
12+
#SBATCH --account=${account}
13+
% endif
14+
% if partition:
15+
#SBATCH --partition=${partition}
16+
% endif
17+
% if quality_of_service:
18+
#SBATCH --qos=${quality_of_service}
19+
% endif
20+
% if gpu:
21+
#SBATCH --gres=gpu:V100:${tasks_per_node}
22+
#SBATCH --mem-per-gpu=16G\
23+
% endif
24+
% if email:
25+
#SBATCH --mail-user=${email}
26+
#SBATCH --mail-type="BEGIN, END, FAIL"
27+
% endif
28+
% endif
29+
30+
${helpers.template_prologue()}
31+
32+
ok ":) Loading modules:\n"
33+
cd "${MFC_ROOT_DIR}"
34+
. ./mfc.sh load -c p -m ${'g' if gpu else 'c'}
35+
cd - > /dev/null
36+
echo
37+
38+
% for target in targets:
39+
${helpers.run_prologue(target)}
40+
41+
% if not mpi:
42+
(set -x; ${profiler} "${target.get_install_binpath(case)}")
43+
% else:
44+
mkdir -p /storage/scratch1/6/sbryngelson3/mytmp
45+
chmod 777 /storage/scratch1/6/sbryngelson3/mytmp
46+
(set -x; ${profiler} \
47+
mpirun --mca orte_tmpdir_base /storage/scratch1/6/sbryngelson3/mytmp \
48+
--np ${nodes*tasks_per_node} \
49+
--bind-to none \
50+
"${target.get_install_binpath(case)}")
51+
% endif
52+
53+
${helpers.run_epilogue(target)}
54+
55+
echo
56+
% endfor
57+
58+
${helpers.template_epilogue()}

0 commit comments

Comments
 (0)