From c523597e111e78bb65adad8ab85ef72e00fcde92 Mon Sep 17 00:00:00 2001
From: Mikko Hakala <mikko.hakala@aalto.fi>
Date: Wed, 1 Dec 2021 16:09:24 +0200
Subject: [PATCH] Address config changes for 21.08+

---
 defaults/main.yml        | 1 +
 templates/cgroup.conf.j2 | 2 ++
 templates/slurm.conf.j2  | 2 ++
 3 files changed, 5 insertions(+)

diff --git a/defaults/main.yml b/defaults/main.yml
index e1ef73c..d513e71 100644
--- a/defaults/main.yml
+++ b/defaults/main.yml
@@ -192,6 +192,7 @@ slurm_JobAcctGatherType: "jobacct_gather/linux"
 slurm_JobAcctGatherFrequency: "energy=60,task=60"
 slurm_AcctGatherEnergyType: "acct_gather_energy/rapl"
 slurm_AcctGatherNodeFreq: "60"
+slurm_accounting_store_flags: "job_comment"
 
 # Job-private /tmp, /var/tmp and /dev/shm
 pam_use_namespace: False
diff --git a/templates/cgroup.conf.j2 b/templates/cgroup.conf.j2
index 5c8a468..279e414 100644
--- a/templates/cgroup.conf.j2
+++ b/templates/cgroup.conf.j2
@@ -11,7 +11,9 @@ CgroupMountpoint={{ slurm_cgroup_mountpoint }}
 CgroupAutomount={{ slurm_cgroup_automount }}
 
 ConstrainCores={{ slurm_cgroup_constrain_cores }}
+{% if slurm_major_version|int < 21 %}
 TaskAffinity={{ slurm_cgroup_taskaffinity }}
+{% endif %}
 {% if slurm_cgroup_constrain_kmem_space is defined %}
 ConstrainKmemSpace={{ slurm_cgroup_constrain_kmem_space }}
 {% endif %}
diff --git a/templates/slurm.conf.j2 b/templates/slurm.conf.j2
index 4fab0f2..744bcb8 100644
--- a/templates/slurm.conf.j2
+++ b/templates/slurm.conf.j2
@@ -103,6 +103,8 @@ AccountingStorageType={{ slurm_accounting_storage_type }}
 AccountingStorageHost={{ slurm_accounting_storage_host }}
 {% if slurm_major_version|int < 21 %}
 AccountingStorageLoc={{ slurm_accounting_storage_loc }}
+{% else %}
+AccountingStoreFlags={{ slurm_accounting_store_flags }}
 {% endif %}
 AccountingStorageUser={{ slurm_accounting_storage_user }}
 AccountingStorageEnforce={{ slurm_accounting_storage_enforce }}