Skip to content

Commit 04d8a0c

Browse files
committed
TBG Bash: up to 8 GPUs
Increase the GPUs per node to 8 in "bash" profile for TBG. Usually, one would do one-node runs with those anyway or would need to adjust it by hand of multi-node is desired. With this setting, using a full 8-GPU DGX-1 node is possible without hacks.
1 parent ff87a58 commit 04d8a0c

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

etc/picongpu/bash/mpiexec.tpl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,8 @@
2424
.TBG_author=${MY_NAME:+--author \"${MY_NAME}\"}
2525
.TBG_profile=${PIC_PROFILE:-"~/picongpu.profile"}
2626

27-
# 4 gpus per node if we need more than 4 gpus else same count as TBG_tasks
28-
.TBG_gpusPerNode=$(if [ $TBG_tasks -gt 4 ] ; then echo 4; else echo $TBG_tasks; fi)
27+
# 8 gpus per node if we need more than 8 gpus else same count as TBG_tasks
28+
.TBG_gpusPerNode=$(if [ $TBG_tasks -gt 8 ] ; then echo 8; else echo $TBG_tasks; fi)
2929

3030
## end calculations ##
3131

etc/picongpu/bash/mpirun.tpl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,8 @@
2424
.TBG_author=${MY_NAME:+--author \"${MY_NAME}\"}
2525
.TBG_profile=${PIC_PROFILE:-"~/picongpu.profile"}
2626

27-
# 4 gpus per node if we need more than 4 gpus else same count as TBG_tasks
28-
.TBG_gpusPerNode=$(if [ $TBG_tasks -gt 4 ] ; then echo 4; else echo $TBG_tasks; fi)
27+
# 8 gpus per node if we need more than 8 gpus else same count as TBG_tasks
28+
.TBG_gpusPerNode=$(if [ $TBG_tasks -gt 8 ] ; then echo 8; else echo $TBG_tasks; fi)
2929

3030
## end calculations ##
3131

0 commit comments

Comments
 (0)