Skip to content

Commit 5f61a8a

Browse files
committed
Bump to NVML version 12.5
Signed-off-by: Kevin Klues <[email protected]>
1 parent c32d291 commit 5f61a8a

File tree

11 files changed

+939
-209
lines changed

11 files changed

+939
-209
lines changed

gen/nvml/nvml.h

Lines changed: 215 additions & 89 deletions
Large diffs are not rendered by default.

gen/nvml/nvml.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -96,4 +96,5 @@ TRANSLATOR:
9696
- {action: replace, from: "^nvmlDeviceGetMPSComputeRunningProcesses$", to: "nvmlDeviceGetMPSComputeRunningProcesses_v1"}
9797
- {action: replace, from: "^nvmlDeviceGetGpuInstancePossiblePlacements$", to: "nvmlDeviceGetGpuInstancePossiblePlacements_v1"}
9898
- {action: replace, from: "^nvmlVgpuInstanceGetLicenseInfo$", to: "nvmlVgpuInstanceGetLicenseInfo_v1"}
99+
- {action: replace, from: "^nvmlDeviceGetDriverModel$", to: "nvmlDeviceGetDriverModel_v1"}
99100
- {transform: unexport}

pkg/nvml/const.go

Lines changed: 17 additions & 3 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

pkg/nvml/device.go

Lines changed: 44 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3055,3 +3055,47 @@ func (device nvmlDevice) GetSramEccErrorStatus() (EccSramErrorStatus, Return) {
30553055
ret := nvmlDeviceGetSramEccErrorStatus(device, &status)
30563056
return status, ret
30573057
}
3058+
3059+
// nvml.DeviceGetClockOffsets()
3060+
func (l *library) DeviceGetClockOffsets(device Device) (ClockOffset, Return) {
3061+
return device.GetClockOffsets()
3062+
}
3063+
3064+
func (device nvmlDevice) GetClockOffsets() (ClockOffset, Return) {
3065+
var info ClockOffset
3066+
info.Version = STRUCT_VERSION(info, 1)
3067+
ret := nvmlDeviceGetClockOffsets(device, &info)
3068+
return info, ret
3069+
}
3070+
3071+
// nvml.DeviceSetClockOffsets()
3072+
func (l *library) DeviceSetClockOffsets(device Device, info ClockOffset) Return {
3073+
return device.SetClockOffsets(info)
3074+
}
3075+
3076+
func (device nvmlDevice) SetClockOffsets(info ClockOffset) Return {
3077+
return nvmlDeviceSetClockOffsets(device, &info)
3078+
}
3079+
3080+
// nvml.DeviceGetDriverModel_v2()
3081+
func (l *library) DeviceGetDriverModel_v2(device Device) (DriverModel, DriverModel, Return) {
3082+
return device.GetDriverModel_v2()
3083+
}
3084+
3085+
func (device nvmlDevice) GetDriverModel_v2() (DriverModel, DriverModel, Return) {
3086+
var current, pending DriverModel
3087+
ret := nvmlDeviceGetDriverModel_v2(device, &current, &pending)
3088+
return current, pending, ret
3089+
}
3090+
3091+
// nvml.DeviceGetCapabilities()
3092+
func (l *library) DeviceGetCapabilities(device Device) (DeviceCapabilities, Return) {
3093+
return device.GetCapabilities()
3094+
}
3095+
3096+
func (device nvmlDevice) GetCapabilities() (DeviceCapabilities, Return) {
3097+
var caps DeviceCapabilities
3098+
caps.Version = STRUCT_VERSION(caps, 1)
3099+
ret := nvmlDeviceGetCapabilities(device, &caps)
3100+
return caps, ret
3101+
}

pkg/nvml/lib.go

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -163,6 +163,7 @@ var GetBlacklistDeviceCount = GetExcludedDeviceCount
163163
var GetBlacklistDeviceInfoByIndex = GetExcludedDeviceInfoByIndex
164164
var nvmlDeviceGetGpuInstancePossiblePlacements = nvmlDeviceGetGpuInstancePossiblePlacements_v1
165165
var nvmlVgpuInstanceGetLicenseInfo = nvmlVgpuInstanceGetLicenseInfo_v1
166+
var nvmlDeviceGetDriverModel = nvmlDeviceGetDriverModel_v1
166167

167168
// BlacklistDeviceInfo was replaced by ExcludedDeviceInfo
168169
type BlacklistDeviceInfo = ExcludedDeviceInfo
@@ -288,4 +289,8 @@ func (l *library) updateVersionedSymbols() {
288289
if err == nil {
289290
nvmlVgpuInstanceGetLicenseInfo = nvmlVgpuInstanceGetLicenseInfo_v2
290291
}
292+
err = l.dl.Lookup("nvmlDeviceGetDriverModel_v2")
293+
if err == nil {
294+
nvmlDeviceGetDriverModel = nvmlDeviceGetDriverModel_v2
295+
}
291296
}

0 commit comments

Comments
 (0)