Skip to content

Commit

Permalink
Merge pull request #45 from Nimbus318/fix-gpu-health
Browse files Browse the repository at this point in the history
Make GPU Health comparison case-insensitive
  • Loading branch information
archlitchi authored Feb 18, 2025
2 parents 2e0c4f9 + f119f1a commit d3d3797
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion pkg/plugin/vgpu/register.go
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@ package vgpu

import (
"fmt"
"strings"
"time"

"github.com/NVIDIA/gpu-monitoring-tools/bindings/go/nvml"
Expand Down Expand Up @@ -69,7 +70,7 @@ func (r *DeviceRegister) apiDevices() *[]*util.DeviceInfo {
Count: int32(config.DeviceSplitCount),
Devmem: registeredmem,
Type: fmt.Sprintf("%v-%v", "NVIDIA", *ndev.Model),
Health: dev.Health == "healthy",
Health: strings.EqualFold(dev.Health, "healthy"),
})
}
return &res
Expand Down

0 comments on commit d3d3797

Please sign in to comment.