mirror of
https://github.com/paboyle/Grid.git
synced 2024-11-14 09:45:36 +00:00
29 lines
752 B
Bash
29 lines
752 B
Bash
#!/bin/bash
|
|
# Begin LSF Directives
|
|
#SBATCH -A LGT104
|
|
#SBATCH -t 01:00:00
|
|
##SBATCH -U openmpThu
|
|
#SBATCH -J DWF
|
|
#SBATCH -o DWF.%J
|
|
#SBATCH -e DWF.%J
|
|
#SBATCH -N 1
|
|
#SBATCH -n 8
|
|
##SBATCH --gpu-bind=map_gpu:0,1,2,3,7,6,5,4
|
|
#SBATCH --gpu-bind=map_gpu:0,1,2,3,6,7,4,5
|
|
|
|
DIR=.
|
|
source setup.sh
|
|
|
|
export MPICH_OFI_NIC_POLICY=GPU
|
|
export MPIR_CVAR_GPU_EAGER_DEVICE_MEM=0
|
|
export MPICH_GPU_SUPPORT_ENABLED=1
|
|
#export MPICH_SMP_SINGLE_COPY_MODE=XPMEM
|
|
#export MPICH_SMP_SINGLE_COPY_MODE=CMA
|
|
#export MPICH_SMP_SINGLE_COPY_MODE=NONE
|
|
export OMP_NUM_THREADS=16
|
|
|
|
echo MPICH_SMP_SINGLE_COPY_MODE $MPICH_SMP_SINGLE_COPY_MODE
|
|
|
|
srun --gpus-per-task 1 -N1 -n8 ./benchmarks/Benchmark_comms_host_device --mpi 2.2.2.1 --shm-mpi 1 --shm 2048 --comms-sequential --accelerator-threads 8
|
|
|