Skip to content

Commit

Permalink
some notes
Browse files Browse the repository at this point in the history
  • Loading branch information
josephdviviano committed Aug 2, 2024
1 parent 2786155 commit 79d58f3
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions tutorials/examples/multinode/ddp_gfn.small.4.slurm
Original file line number Diff line number Diff line change
Expand Up @@ -4,10 +4,13 @@
#SBATCH -J ddp
#SBATCH --get-user-env
#SBATCH --partition=spr
#SBATCH --ntasks=2
#SBATCH --cpus-per-task=112
#SBATCH --ntasks=2 # Number of sockets because we use all threads per socket.
#SBATCH --cpus-per-task=112 # Number of threads available per socket.
#SBATCH --time=00:60:00

# In my case (4 numa nodes) - would set ntasks=4 and cpus-per-task=8 because my CPU has
# 32 cores and I want to divide them across the 4 numa nodes.

source /swtools/intel/2024.0/oneapi-vars.sh
export I_MPI_HYDRA_BOOTSTRAP=slurm

Expand Down

0 comments on commit 79d58f3

Please sign in to comment.