From 70d1b3291ecc02a089941b0b5be0eff105e58688 Mon Sep 17 00:00:00 2001 From: Oren Amsalem Date: Tue, 3 Sep 2024 20:42:38 +0000 Subject: [PATCH] change from gres:gpu to gres/gpu --- slurm_gpustat.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/slurm_gpustat.py b/slurm_gpustat.py index b4627d8..7d37dc4 100644 --- a/slurm_gpustat.py +++ b/slurm_gpustat.py @@ -623,7 +623,9 @@ def gpu_usage(resources: dict, partition: Optional[str] = None) -> dict: resource_flag = "gres" else: resource_flag = "tres-per-node" - if int(slurm_version[0:2]) >= 21: + if int(slurm_version[0:2]) >= 23: + gpu_identifier = 'gres/gpu' + elif int(slurm_version[0:2]) >= 21: gpu_identifier = 'gres:gpu' else: gpu_identifier = 'gpu'