Merge pull request #39 from jabl/accounting-tres

Enable setting AccountinStorageTRES
fgci-org · Mar 30, 2016 · 4d16e10 · 4d16e10
2 parents 210eea2 + 404d1ee
commit 4d16e10
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/defaults/main.yml b/defaults/main.yml
@@ -105,6 +105,9 @@ slurm_accounting_storage_host: "{{ slurm_service_node }}"
 slurm_accounting_storage_type: "accounting_storage/slurmdbd"
 slurm_accounting_storage_loc: "slurm_acct_db"
 slurm_accounting_storage_user: "slurm"
+# If you wish to add accounting for additional TRES types, uncomment
+# and edit the following:
+#slurm_accounting_storage_tres: "gres/gpu:teslak80,gres/gpu:teslak40"
 # If you wish to use a topology plugins (parameter TopologyPlugin=),
 # uncomment and edit the following:
 #slurm_topology_plugin: "topology/tree"

diff --git a/templates/slurm.conf.j2 b/templates/slurm.conf.j2
@@ -133,7 +133,11 @@ AccountingStorageLoc={{ slurm_accounting_storage_loc }}
 #AccountingStoragePass=
 AccountingStorageUser={{ slurm_accounting_storage_user }}
 AccountingStorageEnforce={{ slurm_accounting_storage_enforce }}
-
+{% if slurm_accounting_storage_tres is defined %}
+AccuntingStorageTRES={{ slurm_accounting_storage_tres }}
+{% elif slurm_with_gpu %}
+AccountingStorageTRES=gres/gpu:teslak80
+{% endif %}
 
 #
 # TOPOLOGY