2021-09-16 00:00:38 +01:00
|
|
|
#!/bin/bash
|
|
|
|
|
#SBATCH -J dslash
|
2022-02-22 19:58:33 +00:00
|
|
|
#SBATCH -A dp207
|
2021-09-16 00:00:38 +01:00
|
|
|
#SBATCH --exclusive
|
|
|
|
|
#SBATCH --nodes=4
|
|
|
|
|
#SBATCH --ntasks=16
|
2022-02-22 19:58:33 +00:00
|
|
|
#SBATCH --qos=standard
|
2021-09-16 00:00:38 +01:00
|
|
|
#SBATCH --ntasks-per-node=4
|
|
|
|
|
#SBATCH --cpus-per-task=8
|
2022-02-22 19:58:33 +00:00
|
|
|
#SBATCH --time=0:05:00
|
2021-09-16 00:00:38 +01:00
|
|
|
#SBATCH --partition=gpu
|
|
|
|
|
#SBATCH --gres=gpu:4
|
|
|
|
|
#SBATCH --output=%x.%j.out
|
|
|
|
|
#SBATCH --error=%x.%j.err
|
|
|
|
|
|
|
|
|
|
export OMP_NUM_THREADS=4
|
|
|
|
|
export OMPI_MCA_btl=^uct,openib
|
|
|
|
|
export UCX_TLS=gdr_copy,rc,rc_x,sm,cuda_copy,cuda_ipc
|
|
|
|
|
export UCX_RNDV_SCHEME=put_zcopy
|
|
|
|
|
export UCX_RNDV_THRESH=16384
|
|
|
|
|
export UCX_IB_GPU_DIRECT_RDMA=yes
|
|
|
|
|
export UCX_MEMTYPE_CACHE=n
|
|
|
|
|
OPT="--comms-overlap --comms-concurrent"
|
|
|
|
|
|
|
|
|
|
|
2021-09-16 00:15:35 +01:00
|
|
|
mpirun -np $SLURM_NTASKS -x LD_LIBRARY_PATH --bind-to none \
|
|
|
|
|
./mpiwrapper.sh \
|
|
|
|
|
./benchmarks/Benchmark_dwf_fp32 \
|
|
|
|
|
$OPT \
|
|
|
|
|
--mpi 2.2.2.2 \
|
|
|
|
|
--accelerator-threads 8 \
|
|
|
|
|
--grid 64.64.64.64 \
|
2021-09-16 01:06:28 +01:00
|
|
|
--shm 2048 > dwf.4node.perf
|
2021-09-16 00:00:38 +01:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
