From a6a13db7695f86cac72aa9f497504fa150e5b2ea Mon Sep 17 00:00:00 2001
From: jamesbeedy <jamesbeedy@gmail.com>
Date: Mon, 12 Aug 2024 22:02:29 +0000
Subject: [PATCH] add TaskPlugin to default parameters

These changes address #16 by adding sane defaults for the
TaskPlugin configuration parameter as specified in the
[slurm configuration](https://slurm.schedmd.com/slurm.conf.html#OPT_TaskPlugin).
---
 charms/slurmctld/src/charm.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/charms/slurmctld/src/charm.py b/charms/slurmctld/src/charm.py
index c5d34b0..8e75863 100755
--- a/charms/slurmctld/src/charm.py
+++ b/charms/slurmctld/src/charm.py
@@ -311,6 +311,7 @@ def _assemble_slurmctld_parameters() -> str:
             "SlurmctldHost": self.hostname,
             "SlurmctldParameters": _assemble_slurmctld_parameters(),
             "ProctrackType": "proctrack/linuxproc" if is_container() else "proctrack/cgroup",
+            "TaskPlugin": "task/affinity" if is_container() else "task/cgroup,task/affinity",
             **accounting_params,
             **CHARM_MAINTAINED_SLURM_CONF_PARAMETERS,
             **slurmd_parameters,