2022-02-14 20:09:08 +00:00
|
|
|
#!/bin/bash
|
|
|
|
# Begin LSF Directives
|
|
|
|
#SBATCH -A LGT104
|
2022-02-14 22:55:16 +00:00
|
|
|
#SBATCH -t 01:00:00
|
|
|
|
##SBATCH -U openmpThu
|
2022-02-14 20:09:08 +00:00
|
|
|
#SBATCH -J DWF
|
|
|
|
#SBATCH -o DWF.%J
|
|
|
|
#SBATCH -e DWF.%J
|
|
|
|
#SBATCH -N 1
|
|
|
|
#SBATCH -n 8
|
2022-02-14 22:55:16 +00:00
|
|
|
#SBATCH --exclusive
|
2022-02-14 20:09:08 +00:00
|
|
|
|
|
|
|
DIR=.
|
|
|
|
module list
|
|
|
|
export MPIR_CVAR_GPU_EAGER_DEVICE_MEM=0
|
|
|
|
export MPICH_GPU_SUPPORT_ENABLED=1
|
2022-02-15 13:52:07 +00:00
|
|
|
export MPICH_SMP_SINGLE_COPY_MODE=XPMEM
|
|
|
|
#export MPICH_SMP_SINGLE_COPY_MODE=NONE
|
2022-02-14 20:09:08 +00:00
|
|
|
#export MPICH_SMP_SINGLE_COPY_MODE=CMA
|
|
|
|
export OMP_NUM_THREADS=1
|
|
|
|
|
|
|
|
echo MPICH_SMP_SINGLE_COPY_MODE $MPICH_SMP_SINGLE_COPY_MODE
|
2022-02-14 22:55:16 +00:00
|
|
|
PARAMS=" --accelerator-threads 8 --grid 32.64.64.64 --mpi 1.2.2.2 --comms-overlap --shm 2048 --shm-mpi 0"
|
|
|
|
|
|
|
|
srun --gpus-per-task 1 -n8 ./mpiwrapper.sh ./benchmarks/Benchmark_dwf_fp32 $PARAMS
|
2022-02-14 20:09:08 +00:00
|
|
|
|
|
|
|
|