Bump to NVML version 12.1

Signed-off-by: Kevin Klues <kklues@nvidia.com>
NVIDIA · May 24, 2024 · 9fc23a0 · 9fc23a0
1 parent e035a38
commit 9fc23a0
Show file tree

Hide file tree

Showing 9 changed files with 311 additions and 1,182 deletions.
diff --git a/gen/nvml/nvml.h b/gen/nvml/nvml.h
diff --git a/pkg/nvml/const.go b/pkg/nvml/const.go
diff --git a/pkg/nvml/device.go b/pkg/nvml/device.go
@@ -2654,48 +2654,6 @@ func (device nvmlDevice) GetVgpuCapabilities(capability DeviceVgpuCapability) (b
 	return (capResult != 0), ret
 }
 
-// nvml.DeviceGetVgpuSchedulerLog()
-func (l *library) DeviceGetVgpuSchedulerLog(device Device) (VgpuSchedulerLog, Return) {
-	return device.GetVgpuSchedulerLog()
-}
-
-func (device nvmlDevice) GetVgpuSchedulerLog() (VgpuSchedulerLog, Return) {
-	var pSchedulerLog VgpuSchedulerLog
-	ret := nvmlDeviceGetVgpuSchedulerLog(device, &pSchedulerLog)
-	return pSchedulerLog, ret
-}
-
-// nvml.DeviceGetVgpuSchedulerState()
-func (l *library) DeviceGetVgpuSchedulerState(device Device) (VgpuSchedulerGetState, Return) {
-	return device.GetVgpuSchedulerState()
-}
-
-func (device nvmlDevice) GetVgpuSchedulerState() (VgpuSchedulerGetState, Return) {
-	var pSchedulerState VgpuSchedulerGetState
-	ret := nvmlDeviceGetVgpuSchedulerState(device, &pSchedulerState)
-	return pSchedulerState, ret
-}
-
-// nvml.DeviceSetVgpuSchedulerState()
-func (l *library) DeviceSetVgpuSchedulerState(device Device, pSchedulerState *VgpuSchedulerSetState) Return {
-	return device.SetVgpuSchedulerState(pSchedulerState)
-}
-
-func (device nvmlDevice) SetVgpuSchedulerState(pSchedulerState *VgpuSchedulerSetState) Return {
-	return nvmlDeviceSetVgpuSchedulerState(device, pSchedulerState)
-}
-
-// nvml.DeviceGetVgpuSchedulerCapabilities()
-func (l *library) DeviceGetVgpuSchedulerCapabilities(device Device) (VgpuSchedulerCapabilities, Return) {
-	return device.GetVgpuSchedulerCapabilities()
-}
-
-func (device nvmlDevice) GetVgpuSchedulerCapabilities() (VgpuSchedulerCapabilities, Return) {
-	var pCapabilities VgpuSchedulerCapabilities
-	ret := nvmlDeviceGetVgpuSchedulerCapabilities(device, &pCapabilities)
-	return pCapabilities, ret
-}
-
 // nvml.GpuInstanceGetComputeInstancePossiblePlacements()
 func (l *library) GpuInstanceGetComputeInstancePossiblePlacements(gpuInstance GpuInstance, info *ComputeInstanceProfileInfo) ([]ComputeInstancePlacement, Return) {
 	return gpuInstance.GetComputeInstancePossiblePlacements(info)

diff --git a/pkg/nvml/mock/device.go b/pkg/nvml/mock/device.go