From 54f536c34248f1816e3d8745200c72b4689f510c Mon Sep 17 00:00:00 2001
From: "McCarthy, Samuel (ITS Product Mgmt)" <s.mccarthy@surrey.ac.uk>
Date: Tue, 10 Sep 2024 16:08:04 +0000
Subject: [PATCH] Update file slurm_gpustat.py

---
 slurm_gpustat/slurm_gpustat.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/slurm_gpustat/slurm_gpustat.py b/slurm_gpustat/slurm_gpustat.py
index f3566a0..7d10257 100644
--- a/slurm_gpustat/slurm_gpustat.py
+++ b/slurm_gpustat/slurm_gpustat.py
@@ -547,7 +547,7 @@ def gpu_usage_grouped_by_user(resources: dict, partition: Optional[str] = None)
     if slurm_version.startswith("17"):
         resource_flag = "gres"
     else:
-        resource_flag = "tres-per-node"
+        resource_flag = "tres-per-job"
     cmd = f"squeue -O {resource_flag}:100,nodelist:100,username:100,jobid:100 --noheader"
     if partition:
         cmd += f" --partition={partition}"
-- 
GitLab