diff options
Diffstat (limited to '')
| -rw-r--r-- | Tools/machines/perlmutter-nersc/perlmutter.sbatch | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/Tools/machines/perlmutter-nersc/perlmutter.sbatch b/Tools/machines/perlmutter-nersc/perlmutter.sbatch index 2c085364d..65777f304 100644 --- a/Tools/machines/perlmutter-nersc/perlmutter.sbatch +++ b/Tools/machines/perlmutter-nersc/perlmutter.sbatch @@ -16,8 +16,7 @@ #SBATCH -C gpu #SBATCH -c 32 #SBATCH --ntasks-per-node=4 -#SBATCH --gpus-per-task=1 -#SBATCH --gpu-bind=single:1 +#SBATCH --gpus-per-node=4 #SBATCH -o WarpX.o%j #SBATCH -e WarpX.e%j @@ -42,6 +41,9 @@ # GPU-aware MPI export MPICH_GPU_SUPPORT_ENABLED=1 +# expose one GPU per MPI rank +export CUDA_VISIBLE_DEVICES=$SLURM_LOCALID + EXE=./warpx #EXE=../WarpX/build/bin/warpx.3d.MPI.CUDA.DP.OPMD.QED #EXE=./main3d.gnu.TPROF.MPI.CUDA.ex |