Skip to content

Commit 7685110

Browse files
authored
Merge pull request #62 from klueska/add-archs
Add new ARCH strings from latest NVML
2 parents ecc09d7 + 5b024ba commit 7685110

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

pkg/nvlib/device/device.go

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,10 @@ func (d *device) GetArchitectureAsString() (string, error) {
8989
return "Ada Lovelace", nil
9090
case nvml.DEVICE_ARCH_HOPPER:
9191
return "Hopper", nil
92+
case nvml.DEVICE_ARCH_BLACKWELL:
93+
return "Blackwell", nil
94+
case nvml.DEVICE_ARCH_T23X:
95+
return "Orin", nil
9296
case nvml.DEVICE_ARCH_UNKNOWN:
9397
return "Unknown", nil
9498
}

0 commit comments

Comments
 (0)