Skip to content

Commit

Permalink
L4 does not support FP64
Browse files Browse the repository at this point in the history
  • Loading branch information
guilbaults committed Apr 17, 2024
1 parent 622d185 commit 33fc872
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion slurm-job-exporter.py
Original file line number Diff line number Diff line change
Expand Up @@ -161,7 +161,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 ['NVIDIA RTX A6000']:
if name in ['NVIDIA RTX A6000', 'NVIDIA L4']:
# 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]
Expand Down

0 comments on commit 33fc872

Please sign in to comment.