Skip to content

Commit f7dc5f1

Browse files
authored
Merge pull request #1106 from NVIDIA/gpu-mode-labeler-optional
Raise a warning instead of an error if GPU mode labeler fails
2 parents 61887fe + bd00e7d commit f7dc5f1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

internal/lm/nvml.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -215,7 +215,8 @@ func isMPSCapable(manager resource.Manager) (bool, error) {
215215
func newGPUModeLabeler(devices []resource.Device) (Labeler, error) {
216216
classes, err := getDeviceClasses(devices)
217217
if err != nil {
218-
return nil, err
218+
klog.Warningf("Failed to create GPU mode labeler: failed to get device classes: %v", err)
219+
return Labels{"nvidia.com/gpu.mode": "unknown"}, nil
219220
}
220221
gpuMode := getModeForClasses(classes)
221222
labels := Labels{

0 commit comments

Comments
 (0)