From 1c1d3f32310871b95170bbb4b400309e5ba8cb89 Mon Sep 17 00:00:00 2001
From: Rodrigo Arias Mallo <rodrigo.arias@bsc.es>
Date: Wed, 21 Jun 2023 13:16:23 +0200
Subject: [PATCH] Allow srun to specify the cpu binding

The task/affinity plugin needs to be selected.
---
 m/common/slurm.nix | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/m/common/slurm.nix b/m/common/slurm.nix
index ce909df..08de3fd 100644
--- a/m/common/slurm.nix
+++ b/m/common/slurm.nix
@@ -9,6 +9,8 @@
       "owl[1,2]  Sockets=2 CoresPerSocket=14 ThreadsPerCore=2 Feature=owl"
       "hut       Sockets=2 CoresPerSocket=14 ThreadsPerCore=2"
     ];
+
+    # See slurm.conf(5) for more details about these options.
     extraConfig = ''
       # 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
@@ -20,6 +22,13 @@
       # When a node reboots return that node to the slurm queue as soon as it
       # becomes operative again.
       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
     '';
   };
 }