From b84367f6b4670a2d48bdc3ef93946b646a74e835 Mon Sep 17 00:00:00 2001
From: chn <chn@chn.moe>
Date: Fri, 11 Oct 2024 19:42:37 +0800
Subject: [PATCH] modules.services.slurm: enable task plugins

---
 modules/services/slurm.nix | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/modules/services/slurm.nix b/modules/services/slurm.nix
index a02970bb..03cd86ca 100644
--- a/modules/services/slurm.nix
+++ b/modules/services/slurm.nix
@@ -122,6 +122,9 @@ inputs:
 
             # automatically resume node after drain
             ReturnToService=2
+
+            # enable task plugins
+            TaskPlugin=task/affinity,task/cgroup
           '';
           extraConfigPaths =
             let gpus = slurm.node.${inputs.config.nixos.system.networking.hostname}.gpus or null;