rarias/jungle
rarias/jungle
1
0
Fork 0
Code Issues 27 Pull Requests Packages Projects Releases Wiki Activity

Allow srun to specify the cpu binding

Browse Source 
The task/affinity plugin needs to be selected.
This commit is contained in:
Rodrigo Arias 2023-06-21 13:16:23 +02:00
parent 623d46c03f
commit 1c1d3f3231
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
m/common/slurm.nix
View File

@ -9,6 +9,8 @@
"owl[1,2] Sockets=2 CoresPerSocket=14 ThreadsPerCore=2 Feature=owl" "owl[1,2] Sockets=2 CoresPerSocket=14 ThreadsPerCore=2 Feature=owl"
"hut Sockets=2 CoresPerSocket=14 ThreadsPerCore=2" "hut Sockets=2 CoresPerSocket=14 ThreadsPerCore=2"
]; ];
# See slurm.conf(5) for more details about these options.
extraConfig = '' extraConfig = ''
# Use PMIx for MPI by default. It works okay with MPICH and OpenMPI, but # Use PMIx for MPI by default. It works okay with MPICH and OpenMPI, but
# not with Intel MPI. For that use the compatibility shim libpmi.so # not with Intel MPI. For that use the compatibility shim libpmi.so
@ -20,6 +22,13 @@
# When a node reboots return that node to the slurm queue as soon as it # When a node reboots return that node to the slurm queue as soon as it
# becomes operative again. # becomes operative again.
ReturnToService=2 ReturnToService=2
# Track all processes by using a cgroup
ProctrackType=proctrack/cgroup
# Enable task/affinity to allow the jobs to run in a specified subset of
# the resources. Use the task/cgroup plugin to enable process containment.
TaskPlugin=task/affinity,task/cgroup
''; '';
}; };
} }