Skip to content

Commit 62ce1e3

Browse files
committed
Bump to NVML version 12.9
Signed-off-by: Kevin Klues <[email protected]>
1 parent f420a32 commit 62ce1e3

File tree

14 files changed

+3498
-266
lines changed

14 files changed

+3498
-266
lines changed

examples/gpm-metrics/main.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ func collectGPMMetrics(i int) error {
8989
NumMetrics: 1,
9090
Sample1: sample1,
9191
Sample2: sample2,
92-
Metrics: [98]nvml.GpmMetric{
92+
Metrics: [210]nvml.GpmMetric{
9393
{
9494
MetricId: uint32(nvml.GPM_METRIC_GRAPHICS_UTIL),
9595
},

gen/nvml/nvml.h

Lines changed: 911 additions & 81 deletions
Large diffs are not rendered by default.

pkg/nvml/const.go

Lines changed: 260 additions & 92 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: 97 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -126,6 +126,13 @@ func (l *library) DeviceGetHandleByUUID(uuid string) (Device, Return) {
126126
return device, ret
127127
}
128128

129+
// nvml.DeviceGetHandleByUUIDV()
130+
func (l *library) DeviceGetHandleByUUIDV(uuid *UUID) (Device, Return) {
131+
var device nvmlDevice
132+
ret := nvmlDeviceGetHandleByUUIDV(uuid, &device)
133+
return device, ret
134+
}
135+
129136
// nvml.DeviceGetHandleByPciBusId()
130137
func (l *library) DeviceGetHandleByPciBusId(pciBusId string) (Device, Return) {
131138
var device nvmlDevice
@@ -3315,3 +3322,93 @@ func (l *library) DevicePowerSmoothingSetState(device Device, state *PowerSmooth
33153322
func (device nvmlDevice) PowerSmoothingSetState(state *PowerSmoothingState) Return {
33163323
return nvmlDevicePowerSmoothingSetState(device, state)
33173324
}
3325+
3326+
// nvml.GpuInstanceGetCreatableVgpus()
3327+
func (l *library) GpuInstanceGetCreatableVgpus(gpuInstance GpuInstance) (VgpuTypeIdInfo, Return) {
3328+
return gpuInstance.GetCreatableVgpus()
3329+
}
3330+
3331+
func (gpuInstance nvmlGpuInstance) GetCreatableVgpus() (VgpuTypeIdInfo, Return) {
3332+
var vgpuTypeIdInfo VgpuTypeIdInfo
3333+
vgpuTypeIdInfo.Version = STRUCT_VERSION(vgpuTypeIdInfo, 1)
3334+
ret := nvmlGpuInstanceGetCreatableVgpus(gpuInstance, &vgpuTypeIdInfo)
3335+
return vgpuTypeIdInfo, ret
3336+
}
3337+
3338+
// nvml.GpuInstanceGetActiveVgpus()
3339+
func (l *library) GpuInstanceGetActiveVgpus(gpuInstance GpuInstance) (ActiveVgpuInstanceInfo, Return) {
3340+
return gpuInstance.GetActiveVgpus()
3341+
}
3342+
3343+
func (gpuInstance nvmlGpuInstance) GetActiveVgpus() (ActiveVgpuInstanceInfo, Return) {
3344+
var activeVgpuInstanceInfo ActiveVgpuInstanceInfo
3345+
activeVgpuInstanceInfo.Version = STRUCT_VERSION(activeVgpuInstanceInfo, 1)
3346+
ret := nvmlGpuInstanceGetActiveVgpus(gpuInstance, &activeVgpuInstanceInfo)
3347+
return activeVgpuInstanceInfo, ret
3348+
}
3349+
3350+
// nvml.GpuInstanceSetVgpuSchedulerState()
3351+
func (l *library) GpuInstanceSetVgpuSchedulerState(gpuInstance GpuInstance, scheduler *VgpuSchedulerState) Return {
3352+
return gpuInstance.SetVgpuSchedulerState(scheduler)
3353+
}
3354+
3355+
func (gpuInstance nvmlGpuInstance) SetVgpuSchedulerState(scheduler *VgpuSchedulerState) Return {
3356+
return nvmlGpuInstanceSetVgpuSchedulerState(gpuInstance, scheduler)
3357+
}
3358+
3359+
// nvml.GpuInstanceGetVgpuSchedulerState()
3360+
func (l *library) GpuInstanceGetVgpuSchedulerState(gpuInstance GpuInstance) (VgpuSchedulerStateInfo, Return) {
3361+
return gpuInstance.GetVgpuSchedulerState()
3362+
}
3363+
3364+
func (gpuInstance nvmlGpuInstance) GetVgpuSchedulerState() (VgpuSchedulerStateInfo, Return) {
3365+
var schedulerStateInfo VgpuSchedulerStateInfo
3366+
schedulerStateInfo.Version = STRUCT_VERSION(schedulerStateInfo, 1)
3367+
ret := nvmlGpuInstanceGetVgpuSchedulerState(gpuInstance, &schedulerStateInfo)
3368+
return schedulerStateInfo, ret
3369+
}
3370+
3371+
// nvml.GpuInstanceGetVgpuSchedulerLog()
3372+
func (l *library) GpuInstanceGetVgpuSchedulerLog(gpuInstance GpuInstance) (VgpuSchedulerLogInfo, Return) {
3373+
return gpuInstance.GetVgpuSchedulerLog()
3374+
}
3375+
3376+
func (gpuInstance nvmlGpuInstance) GetVgpuSchedulerLog() (VgpuSchedulerLogInfo, Return) {
3377+
var schedulerLogInfo VgpuSchedulerLogInfo
3378+
schedulerLogInfo.Version = STRUCT_VERSION(schedulerLogInfo, 1)
3379+
ret := nvmlGpuInstanceGetVgpuSchedulerLog(gpuInstance, &schedulerLogInfo)
3380+
return schedulerLogInfo, ret
3381+
}
3382+
3383+
// nvml.GpuInstanceGetVgpuTypeCreatablePlacements()
3384+
func (l *library) GpuInstanceGetVgpuTypeCreatablePlacements(gpuInstance GpuInstance) (VgpuCreatablePlacementInfo, Return) {
3385+
return gpuInstance.GetVgpuTypeCreatablePlacements()
3386+
}
3387+
3388+
func (gpuInstance nvmlGpuInstance) GetVgpuTypeCreatablePlacements() (VgpuCreatablePlacementInfo, Return) {
3389+
var creatablePlacementInfo VgpuCreatablePlacementInfo
3390+
creatablePlacementInfo.Version = STRUCT_VERSION(creatablePlacementInfo, 1)
3391+
ret := nvmlGpuInstanceGetVgpuTypeCreatablePlacements(gpuInstance, &creatablePlacementInfo)
3392+
return creatablePlacementInfo, ret
3393+
}
3394+
3395+
// nvml.GpuInstanceGetVgpuHeterogeneousMode()
3396+
func (l *library) GpuInstanceGetVgpuHeterogeneousMode(gpuInstance GpuInstance) (VgpuHeterogeneousMode, Return) {
3397+
return gpuInstance.GetVgpuHeterogeneousMode()
3398+
}
3399+
3400+
func (gpuInstance nvmlGpuInstance) GetVgpuHeterogeneousMode() (VgpuHeterogeneousMode, Return) {
3401+
var heterogeneousMode VgpuHeterogeneousMode
3402+
heterogeneousMode.Version = STRUCT_VERSION(heterogeneousMode, 1)
3403+
ret := nvmlGpuInstanceGetVgpuHeterogeneousMode(gpuInstance, &heterogeneousMode)
3404+
return heterogeneousMode, ret
3405+
}
3406+
3407+
// nvml.GpuInstanceSetVgpuHeterogeneousMode()
3408+
func (l *library) GpuInstanceSetVgpuHeterogeneousMode(gpuInstance GpuInstance, heterogeneousMode *VgpuHeterogeneousMode) Return {
3409+
return gpuInstance.SetVgpuHeterogeneousMode(heterogeneousMode)
3410+
}
3411+
3412+
func (gpuInstance nvmlGpuInstance) SetVgpuHeterogeneousMode(heterogeneousMode *VgpuHeterogeneousMode) Return {
3413+
return nvmlGpuInstanceSetVgpuHeterogeneousMode(gpuInstance, heterogeneousMode)
3414+
}

pkg/nvml/event_set.go

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,3 +60,23 @@ func (l *library) EventSetFree(set EventSet) Return {
6060
func (set nvmlEventSet) Free() Return {
6161
return nvmlEventSetFree(set)
6262
}
63+
64+
// nvml.SystemEventSetCreate()
65+
func (l *library) SystemEventSetCreate(request *SystemEventSetCreateRequest) Return {
66+
return nvmlSystemEventSetCreate(request)
67+
}
68+
69+
// nvml.SystemEventSetFree()
70+
func (l *library) SystemEventSetFree(request *SystemEventSetFreeRequest) Return {
71+
return nvmlSystemEventSetFree(request)
72+
}
73+
74+
// nvml.SystemRegisterEvents()
75+
func (l *library) SystemRegisterEvents(request *SystemRegisterEventRequest) Return {
76+
return nvmlSystemRegisterEvents(request)
77+
}
78+
79+
// nvml.SystemEventSetWait()
80+
func (l *library) SystemEventSetWait(request *SystemEventSetWaitRequest) Return {
81+
return nvmlSystemEventSetWait(request)
82+
}

pkg/nvml/gpm.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ type GpmMetricsGetType struct {
2020
NumMetrics uint32
2121
Sample1 GpmSample
2222
Sample2 GpmSample
23-
Metrics [98]GpmMetric
23+
Metrics [210]GpmMetric
2424
}
2525

2626
func (g *GpmMetricsGetType) convert() *nvmlGpmMetricsGetType {

pkg/nvml/gpm_test.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ import (
2323
)
2424

2525
func TestGpmMetricsGet(t *testing.T) {
26-
overrideMetrics := [98]GpmMetric{
26+
overrideMetrics := [210]GpmMetric{
2727
{
2828
Value: 99,
2929
},
@@ -46,7 +46,7 @@ func TestGpmMetricsGet(t *testing.T) {
4646
}
4747

4848
func TestGpmMetricsGetV(t *testing.T) {
49-
overrideMetrics := [98]GpmMetric{
49+
overrideMetrics := [210]GpmMetric{
5050
{
5151
Value: 99,
5252
},

0 commit comments

Comments
 (0)