Skip to content

Commit

Permalink
Adding GFD node labels to specify GPU mode
Browse files Browse the repository at this point in the history
Signed-off-by: Vishesh Tanksale <vtanksale@nvidia.com>
  • Loading branch information
visheshtanksale committed Jun 14, 2024
1 parent d92b786 commit 08524f3
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions internal/resource/nvml-device.go
Original file line number Diff line number Diff line change
Expand Up @@ -115,11 +115,10 @@ func resolvePCIAddressToMode(addr string) (string, error) {
}
classStr := string(class)
classStr = strings.TrimSpace(classStr)

switch classStr {
case "0x030000":
return "display", nil
case "0X030200":
case "0x030200":
return "compute", nil
default:
return "unknown", nil
Expand Down

0 comments on commit 08524f3

Please sign in to comment.