Skip to content

Commit 3179be7

Browse files
jaguilarjaguilar
authored andcommitted
Fix tuo mako file to support cpu build and batch scripts.
Switched "gpu" to "gpu_enabled" so that the batch script is properly made when running on CPU and GPU. With just "gpu", it defaults to always being gpu.
1 parent 23bbd4f commit 3179be7

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

toolchain/templates/tuo.mako

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,12 +24,12 @@ ${helpers.template_prologue()}
2424
ok ":) Loading modules:\n"
2525
cd "${MFC_ROOT_DIR}"
2626
% if engine == 'batch':
27-
. ./mfc.sh load -c tuo -m ${'g' if gpu else 'c'}
27+
. ./mfc.sh load -c tuo -m ${'g' if gpu_enabled else 'c'}
2828
% endif
2929
cd - > /dev/null
3030
echo
3131

32-
% if gpu:
32+
% if gpu_enabled:
3333
export MPICH_GPU_SUPPORT_ENABLED=1
3434
% else:
3535
export MPICH_GPU_SUPPORT_ENABLED=0
@@ -46,7 +46,7 @@ export HSA_XNACK=0
4646
(set -x; flux run \
4747
--nodes=${nodes} --ntasks=${tasks_per_node * nodes} \
4848
-o spindle.level=off --exclusive \
49-
% if gpu:
49+
% if gpu_enabled:
5050
--gpus-per-task 1 \
5151
% endif
5252
${profiler} "${target.get_install_binpath(case)}")

0 commit comments

Comments
 (0)