Skip to content

Commit 0f83387

Browse files
dgvacareveloDiego Vaca
andauthored
Turing WPI support (#1335)
Co-authored-by: Diego Vaca <dgvacarevelo@login-01.cm.cluster>
1 parent 2d15ba9 commit 0f83387

3 files changed

Lines changed: 57 additions & 1 deletion

File tree

toolchain/bootstrap/modules.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,8 @@ if [ -v $u_c ]; then
4646
log "$BR""Brown$W: Oscar (o)"
4747
log "$B""DoD$W: Carpenter Cray (cc) | Carpenter GNU (c) | Nautilus (n)"
4848
log "$OR""Florida$W: HiPerGator (h)"
49-
log_n "($G""a$W/$G""f$W/$G""s$W/$G""w$W/$B""tuo$W/$C""b$W/$C""e$CR/$C""d/$C""dai$CR/$Y""p$CR/$R""r$CR/$B""cc$CR/$B""c$CR/$B""n$CR/$BR""o"$CR"/$OR""h"$CR"): "
49+
log "$C""WPI $W: Turing (t)"
50+
log_n "($G""a$W/$G""f$W/$G""s$W/$G""w$W/$B""tuo$W/$C""b$W/$C""e$CR/$C""d/$C""dai$CR/$Y""p$CR/$R""r$CR/$B""cc$CR/$B""c$CR/$B""n$CR/$BR""o"$CR"/$OR""h"$CR/$C""t""$CR"): "
5051
read u_c
5152
log
5253
fi

toolchain/modules

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -102,3 +102,7 @@ h-gpu CC=/apps/compilers/nvhpc/25.9/Linux_x86_64/25.9/comm_libs/mpi/bin/mpicc
102102
h-gpu CXX=/apps/compilers/nvhpc/25.9/Linux_x86_64/25.9/comm_libs/mpi/bin/mpicxx
103103
h-gpu FC=/apps/compilers/nvhpc/25.9/Linux_x86_64/25.9/comm_libs/mpi/bin/mpifort
104104
h-gpu NVCOMPILER_COMM_LIBS_HOME=/apps/compilers/nvhpc/25.9/Linux_x86_64/25.9/comm_libs/12.9
105+
106+
t WPI Turing
107+
t-all slurm
108+
t-cpu gcc/12.1.0/i6yk33f openmpi/4.1.3/ebae7zc python/3.13.5/6anz4qy

toolchain/templates/turing.mako

Lines changed: 51 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,51 @@
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 --cpus-per-task=1
9+
#SBATCH --job-name="${name}"
10+
#SBATCH --time=${walltime}
11+
% if partition:
12+
#SBATCH --partition=${partition}
13+
% endif
14+
% if account:
15+
#SBATCH --account="${account}"
16+
% endif
17+
#SBATCH --output="${name}.out"
18+
#SBATCH --error="${name}.err"
19+
#SBATCH --export=ALL
20+
% if email:
21+
#SBATCH --mail-user=${email}
22+
#SBATCH --mail-type=BEGIN,END,FAIL
23+
% endif
24+
% endif
25+
26+
${helpers.template_prologue()}
27+
28+
ok ":) Loading modules:\n"
29+
cd "${MFC_ROOT_DIR}"
30+
. ./mfc.sh load -c t -m ${'g' if gpu_enabled else 'c'}
31+
cd - > /dev/null
32+
echo
33+
34+
35+
% for target in targets:
36+
${helpers.run_prologue(target)}
37+
38+
% if not mpi:
39+
(set -x; ${profiler} "${target.get_install_binpath(case)}")
40+
% else:
41+
(set -x; ${profiler} \
42+
srun --mpi=pmi2 --ntasks=${nodes*tasks_per_node} \
43+
"${target.get_install_binpath(case)}")
44+
% endif
45+
46+
${helpers.run_epilogue(target)}
47+
48+
echo
49+
% endfor
50+
51+
${helpers.template_epilogue()}

0 commit comments

Comments
 (0)