From 042b191c3c17fb283ca7ae74265ce6f65914321c Mon Sep 17 00:00:00 2001 From: Simon Guilbault Date: Thu, 23 Nov 2023 09:19:24 -0500 Subject: [PATCH] Adding support for NVIDIA RTX A6000 Co-authored-by: Bruno Travouillon --- slurm-job-exporter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/slurm-job-exporter.py b/slurm-job-exporter.py index de31732..9d40018 100644 --- a/slurm-job-exporter.py +++ b/slurm-job-exporter.py @@ -155,7 +155,7 @@ def __init__(self, dcgm_update_interval=10): device = pydcgm.dcgm_agent.dcgmGetDeviceAttributes(self.handle.handle, gpu_id) name = device.identifiers.deviceName print('Detected gpu {} with ID {}'.format(name, gpu_id)) - if name in ['Quadro RTX 6000']: + if name in ['NVIDIA RTX A6000']: # This GPU does not supports fp64, we don't support a mix of fp64 and non-fp64 GPUs in the same node print('Removing fp64 metrics since {} does not support fp64'.format(name)) del self.fieldIds_dict[dcgm_fields.DCGM_FI_PROF_PIPE_FP64_ACTIVE]