mirror of
https://github.com/smyalygames/FiniteVolumeGPU_HIP.git
synced 2025-05-18 06:24:11 +02:00
28 lines
685 B
Bash
28 lines
685 B
Bash
#!/bin/bash -l
|
|
#SBATCH --job-name=lumi
|
|
#SBATCH --account=project_4650000xx
|
|
#SBATCH --time=00:10:00
|
|
#SBATCH --partition=dev-g
|
|
#SBATCH --nodes=1
|
|
#SBATCH --ntasks-per-node=8
|
|
#SBATCH --gpus-per-node=8
|
|
#SBATCH --output=%x-%j.out
|
|
#SBATCH --exclusive
|
|
|
|
N=$SLURM_JOB_NUM_NODES
|
|
echo "--nbr of nodes:", $N
|
|
echo "--total nbr of gpus:", $SLURM_NTASKS
|
|
|
|
Mydir=/project/${project}
|
|
Myapplication=${Mydir}/FiniteVolumeGPU_HIP/mpiTesting.py
|
|
CondaEnv=${Mydir}/FiniteVolumeGPU_HIP/MyCondaEnv/bin
|
|
|
|
export PATH="${CondaEnv}:$PATH"
|
|
|
|
CPU_BIND="map_cpu:49,57,17,25,1,9,33,41"
|
|
|
|
export MPICH_GPU_SUPPORT_ENABLED=1
|
|
|
|
srun --cpu-bind=${CPU_BIND} --mpi=pmi2 \
|
|
python ${Myapplication} -nx 1024 -ny 1024 --profile
|