Bump github.com/NVIDIA/go-nvml from 0.12.0-6 to 0.12.4-0

Bumps [github.com/NVIDIA/go-nvml](https://github.com/NVIDIA/go-nvml) from 0.12.0-6 to 0.12.4-0.
- [Commits](https://github.com/NVIDIA/go-nvml/compare/v0.12.0-6...v0.12.4-0)

---
updated-dependencies:
- dependency-name: github.com/NVIDIA/go-nvml
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
This commit is contained in:
dependabot[bot] 2024-06-14 13:27:46 +00:00 committed by GitHub
parent 55630bc2c0
commit abd638add9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
15 changed files with 6441 additions and 1298 deletions

2
go.mod
View File

@ -4,7 +4,7 @@ go 1.20
require (
github.com/NVIDIA/go-nvlib v0.5.0
github.com/NVIDIA/go-nvml v0.12.0-6
github.com/NVIDIA/go-nvml v0.12.4-0
github.com/fsnotify/fsnotify v1.7.0
github.com/opencontainers/runtime-spec v1.2.0
github.com/pelletier/go-toml v1.9.5

4
go.sum
View File

@ -1,7 +1,7 @@
github.com/NVIDIA/go-nvlib v0.5.0 h1:951KGrfr+p3cs89alO9z/ZxPPWKxwht9tx9rxiADoLI=
github.com/NVIDIA/go-nvlib v0.5.0/go.mod h1:87z49ULPr4GWPSGfSIp3taU4XENRYN/enIg88MzcL4k=
github.com/NVIDIA/go-nvml v0.12.0-6 h1:FJYc2KrpvX+VOC/8QQvMiQMmZ/nPMRpdJO/Ik4xfcr0=
github.com/NVIDIA/go-nvml v0.12.0-6/go.mod h1:8Llmj+1Rr+9VGGwZuRer5N/aCjxGuR5nPb/9ebBiIEQ=
github.com/NVIDIA/go-nvml v0.12.4-0 h1:4tkbB3pT1O77JGr0gQ6uD8FrsUPqP1A/EOEm2wI1TUg=
github.com/NVIDIA/go-nvml v0.12.4-0/go.mod h1:8Llmj+1Rr+9VGGwZuRer5N/aCjxGuR5nPb/9ebBiIEQ=
github.com/blang/semver/v4 v4.0.0 h1:1PFHFE6yCCTv8C1TeyNNarDzntLi7wMI5i/pzqYIsAM=
github.com/blang/semver/v4 v4.0.0/go.mod h1:IbckMUScFkM3pff0VJDNKRiT6TG/YpiHIM2yvyW5YoQ=
github.com/cpuguy83/go-md2man/v2 v2.0.4 h1:wfIWP927BUkWJb2NmU/kNDYIBTh/ziUX91+lVfRxZq4=

View File

@ -31,9 +31,9 @@ const (
// NO_UNVERSIONED_FUNC_DEFS as defined in go-nvml/<predefine>:24
NO_UNVERSIONED_FUNC_DEFS = 1
// API_VERSION as defined in nvml/nvml.h
API_VERSION = 11
API_VERSION = 12
// API_VERSION_STR as defined in nvml/nvml.h
API_VERSION_STR = "11"
API_VERSION_STR = "12"
// VALUE_NOT_AVAILABLE as defined in nvml/nvml.h
VALUE_NOT_AVAILABLE = -1
// DEVICE_PCI_BUS_ID_BUFFER_SIZE as defined in nvml/nvml.h
@ -78,6 +78,8 @@ const (
VGPU_NAME_BUFFER_SIZE = 64
// GRID_LICENSE_FEATURE_MAX_COUNT as defined in nvml/nvml.h
GRID_LICENSE_FEATURE_MAX_COUNT = 3
// INVALID_VGPU_PLACEMENT_ID as defined in nvml/nvml.h
INVALID_VGPU_PLACEMENT_ID = 65535
// VGPU_SCHEDULER_POLICY_UNKNOWN as defined in nvml/nvml.h
VGPU_SCHEDULER_POLICY_UNKNOWN = 0
// VGPU_SCHEDULER_POLICY_BEST_EFFORT as defined in nvml/nvml.h
@ -90,6 +92,12 @@ const (
SUPPORTED_VGPU_SCHEDULER_POLICY_COUNT = 3
// SCHEDULER_SW_MAX_LOG_ENTRIES as defined in nvml/nvml.h
SCHEDULER_SW_MAX_LOG_ENTRIES = 200
// VGPU_SCHEDULER_ARR_DEFAULT as defined in nvml/nvml.h
VGPU_SCHEDULER_ARR_DEFAULT = 0
// VGPU_SCHEDULER_ARR_DISABLE as defined in nvml/nvml.h
VGPU_SCHEDULER_ARR_DISABLE = 1
// VGPU_SCHEDULER_ARR_ENABLE as defined in nvml/nvml.h
VGPU_SCHEDULER_ARR_ENABLE = 2
// GRID_LICENSE_STATE_UNKNOWN as defined in nvml/nvml.h
GRID_LICENSE_STATE_UNKNOWN = 0
// GRID_LICENSE_STATE_UNINITIALIZED as defined in nvml/nvml.h
@ -140,6 +148,8 @@ const (
POWER_SOURCE_AC = 0
// POWER_SOURCE_BATTERY as defined in nvml/nvml.h
POWER_SOURCE_BATTERY = 1
// POWER_SOURCE_UNDERSIZED as defined in nvml/nvml.h
POWER_SOURCE_UNDERSIZED = 2
// PCIE_LINK_MAX_SPEED_INVALID as defined in nvml/nvml.h
PCIE_LINK_MAX_SPEED_INVALID = 0
// PCIE_LINK_MAX_SPEED_2500MBPS as defined in nvml/nvml.h
@ -498,8 +508,64 @@ const (
FI_DEV_NVLINK_GET_POWER_THRESHOLD = 168
// FI_DEV_PCIE_L0_TO_RECOVERY_COUNTER as defined in nvml/nvml.h
FI_DEV_PCIE_L0_TO_RECOVERY_COUNTER = 169
// FI_DEV_C2C_LINK_COUNT as defined in nvml/nvml.h
FI_DEV_C2C_LINK_COUNT = 170
// FI_DEV_C2C_LINK_GET_STATUS as defined in nvml/nvml.h
FI_DEV_C2C_LINK_GET_STATUS = 171
// FI_DEV_C2C_LINK_GET_MAX_BW as defined in nvml/nvml.h
FI_DEV_C2C_LINK_GET_MAX_BW = 172
// FI_DEV_PCIE_COUNT_CORRECTABLE_ERRORS as defined in nvml/nvml.h
FI_DEV_PCIE_COUNT_CORRECTABLE_ERRORS = 173
// FI_DEV_PCIE_COUNT_NAKS_RECEIVED as defined in nvml/nvml.h
FI_DEV_PCIE_COUNT_NAKS_RECEIVED = 174
// FI_DEV_PCIE_COUNT_RECEIVER_ERROR as defined in nvml/nvml.h
FI_DEV_PCIE_COUNT_RECEIVER_ERROR = 175
// FI_DEV_PCIE_COUNT_BAD_TLP as defined in nvml/nvml.h
FI_DEV_PCIE_COUNT_BAD_TLP = 176
// FI_DEV_PCIE_COUNT_NAKS_SENT as defined in nvml/nvml.h
FI_DEV_PCIE_COUNT_NAKS_SENT = 177
// FI_DEV_PCIE_COUNT_BAD_DLLP as defined in nvml/nvml.h
FI_DEV_PCIE_COUNT_BAD_DLLP = 178
// FI_DEV_PCIE_COUNT_NON_FATAL_ERROR as defined in nvml/nvml.h
FI_DEV_PCIE_COUNT_NON_FATAL_ERROR = 179
// FI_DEV_PCIE_COUNT_FATAL_ERROR as defined in nvml/nvml.h
FI_DEV_PCIE_COUNT_FATAL_ERROR = 180
// FI_DEV_PCIE_COUNT_UNSUPPORTED_REQ as defined in nvml/nvml.h
FI_DEV_PCIE_COUNT_UNSUPPORTED_REQ = 181
// FI_DEV_PCIE_COUNT_LCRC_ERROR as defined in nvml/nvml.h
FI_DEV_PCIE_COUNT_LCRC_ERROR = 182
// FI_DEV_PCIE_COUNT_LANE_ERROR as defined in nvml/nvml.h
FI_DEV_PCIE_COUNT_LANE_ERROR = 183
// FI_DEV_IS_RESETLESS_MIG_SUPPORTED as defined in nvml/nvml.h
FI_DEV_IS_RESETLESS_MIG_SUPPORTED = 184
// FI_DEV_POWER_AVERAGE as defined in nvml/nvml.h
FI_DEV_POWER_AVERAGE = 185
// FI_DEV_POWER_INSTANT as defined in nvml/nvml.h
FI_DEV_POWER_INSTANT = 186
// FI_DEV_POWER_MIN_LIMIT as defined in nvml/nvml.h
FI_DEV_POWER_MIN_LIMIT = 187
// FI_DEV_POWER_MAX_LIMIT as defined in nvml/nvml.h
FI_DEV_POWER_MAX_LIMIT = 188
// FI_DEV_POWER_DEFAULT_LIMIT as defined in nvml/nvml.h
FI_DEV_POWER_DEFAULT_LIMIT = 189
// FI_DEV_POWER_CURRENT_LIMIT as defined in nvml/nvml.h
FI_DEV_POWER_CURRENT_LIMIT = 190
// FI_DEV_ENERGY as defined in nvml/nvml.h
FI_DEV_ENERGY = 191
// FI_DEV_POWER_REQUESTED_LIMIT as defined in nvml/nvml.h
FI_DEV_POWER_REQUESTED_LIMIT = 192
// FI_DEV_TEMPERATURE_SHUTDOWN_TLIMIT as defined in nvml/nvml.h
FI_DEV_TEMPERATURE_SHUTDOWN_TLIMIT = 193
// FI_DEV_TEMPERATURE_SLOWDOWN_TLIMIT as defined in nvml/nvml.h
FI_DEV_TEMPERATURE_SLOWDOWN_TLIMIT = 194
// FI_DEV_TEMPERATURE_MEM_MAX_TLIMIT as defined in nvml/nvml.h
FI_DEV_TEMPERATURE_MEM_MAX_TLIMIT = 195
// FI_DEV_TEMPERATURE_GPU_MAX_TLIMIT as defined in nvml/nvml.h
FI_DEV_TEMPERATURE_GPU_MAX_TLIMIT = 196
// FI_DEV_IS_MIG_MODE_INDEPENDENT_MIG_QUERY_CAPABLE as defined in nvml/nvml.h
FI_DEV_IS_MIG_MODE_INDEPENDENT_MIG_QUERY_CAPABLE = 199
// FI_MAX as defined in nvml/nvml.h
FI_MAX = 170
FI_MAX = 200
// EventTypeSingleBitEccError as defined in nvml/nvml.h
EventTypeSingleBitEccError = 1
// EventTypeDoubleBitEccError as defined in nvml/nvml.h
@ -518,24 +584,40 @@ const (
EventTypeNone = 0
// EventTypeAll as defined in nvml/nvml.h
EventTypeAll = 415
// ClocksThrottleReasonGpuIdle as defined in nvml/nvml.h
ClocksThrottleReasonGpuIdle = 1
// ClocksThrottleReasonApplicationsClocksSetting as defined in nvml/nvml.h
ClocksThrottleReasonApplicationsClocksSetting = 2
// ClocksEventReasonGpuIdle as defined in nvml/nvml.h
ClocksEventReasonGpuIdle = 1
// ClocksEventReasonApplicationsClocksSetting as defined in nvml/nvml.h
ClocksEventReasonApplicationsClocksSetting = 2
// ClocksThrottleReasonUserDefinedClocks as defined in nvml/nvml.h
ClocksThrottleReasonUserDefinedClocks = 2
// ClocksThrottleReasonSwPowerCap as defined in nvml/nvml.h
ClocksThrottleReasonSwPowerCap = 4
// ClocksEventReasonSwPowerCap as defined in nvml/nvml.h
ClocksEventReasonSwPowerCap = 4
// ClocksThrottleReasonHwSlowdown as defined in nvml/nvml.h
ClocksThrottleReasonHwSlowdown = 8
// ClocksThrottleReasonSyncBoost as defined in nvml/nvml.h
ClocksThrottleReasonSyncBoost = 16
// ClocksThrottleReasonSwThermalSlowdown as defined in nvml/nvml.h
ClocksThrottleReasonSwThermalSlowdown = 32
// ClocksEventReasonSyncBoost as defined in nvml/nvml.h
ClocksEventReasonSyncBoost = 16
// ClocksEventReasonSwThermalSlowdown as defined in nvml/nvml.h
ClocksEventReasonSwThermalSlowdown = 32
// ClocksThrottleReasonHwThermalSlowdown as defined in nvml/nvml.h
ClocksThrottleReasonHwThermalSlowdown = 64
// ClocksThrottleReasonHwPowerBrakeSlowdown as defined in nvml/nvml.h
ClocksThrottleReasonHwPowerBrakeSlowdown = 128
// ClocksEventReasonDisplayClockSetting as defined in nvml/nvml.h
ClocksEventReasonDisplayClockSetting = 256
// ClocksEventReasonNone as defined in nvml/nvml.h
ClocksEventReasonNone = 0
// ClocksEventReasonAll as defined in nvml/nvml.h
ClocksEventReasonAll = 511
// ClocksThrottleReasonGpuIdle as defined in nvml/nvml.h
ClocksThrottleReasonGpuIdle = 1
// ClocksThrottleReasonApplicationsClocksSetting as defined in nvml/nvml.h
ClocksThrottleReasonApplicationsClocksSetting = 2
// ClocksThrottleReasonSyncBoost as defined in nvml/nvml.h
ClocksThrottleReasonSyncBoost = 16
// ClocksThrottleReasonSwPowerCap as defined in nvml/nvml.h
ClocksThrottleReasonSwPowerCap = 4
// ClocksThrottleReasonSwThermalSlowdown as defined in nvml/nvml.h
ClocksThrottleReasonSwThermalSlowdown = 32
// ClocksThrottleReasonDisplayClockSetting as defined in nvml/nvml.h
ClocksThrottleReasonDisplayClockSetting = 256
// ClocksThrottleReasonNone as defined in nvml/nvml.h
@ -552,6 +634,56 @@ const (
NVFBC_SESSION_FLAG_CAPTURE_WITH_WAIT_INFINITE = 8
// NVFBC_SESSION_FLAG_CAPTURE_WITH_WAIT_TIMEOUT as defined in nvml/nvml.h
NVFBC_SESSION_FLAG_CAPTURE_WITH_WAIT_TIMEOUT = 16
// CC_SYSTEM_CPU_CAPS_NONE as defined in nvml/nvml.h
CC_SYSTEM_CPU_CAPS_NONE = 0
// CC_SYSTEM_CPU_CAPS_AMD_SEV as defined in nvml/nvml.h
CC_SYSTEM_CPU_CAPS_AMD_SEV = 1
// CC_SYSTEM_CPU_CAPS_INTEL_TDX as defined in nvml/nvml.h
CC_SYSTEM_CPU_CAPS_INTEL_TDX = 2
// CC_SYSTEM_GPUS_CC_NOT_CAPABLE as defined in nvml/nvml.h
CC_SYSTEM_GPUS_CC_NOT_CAPABLE = 0
// CC_SYSTEM_GPUS_CC_CAPABLE as defined in nvml/nvml.h
CC_SYSTEM_GPUS_CC_CAPABLE = 1
// CC_SYSTEM_DEVTOOLS_MODE_OFF as defined in nvml/nvml.h
CC_SYSTEM_DEVTOOLS_MODE_OFF = 0
// CC_SYSTEM_DEVTOOLS_MODE_ON as defined in nvml/nvml.h
CC_SYSTEM_DEVTOOLS_MODE_ON = 1
// CC_SYSTEM_ENVIRONMENT_UNAVAILABLE as defined in nvml/nvml.h
CC_SYSTEM_ENVIRONMENT_UNAVAILABLE = 0
// CC_SYSTEM_ENVIRONMENT_SIM as defined in nvml/nvml.h
CC_SYSTEM_ENVIRONMENT_SIM = 1
// CC_SYSTEM_ENVIRONMENT_PROD as defined in nvml/nvml.h
CC_SYSTEM_ENVIRONMENT_PROD = 2
// CC_SYSTEM_FEATURE_DISABLED as defined in nvml/nvml.h
CC_SYSTEM_FEATURE_DISABLED = 0
// CC_SYSTEM_FEATURE_ENABLED as defined in nvml/nvml.h
CC_SYSTEM_FEATURE_ENABLED = 1
// CC_SYSTEM_MULTIGPU_NONE as defined in nvml/nvml.h
CC_SYSTEM_MULTIGPU_NONE = 0
// CC_SYSTEM_MULTIGPU_PROTECTED_PCIE as defined in nvml/nvml.h
CC_SYSTEM_MULTIGPU_PROTECTED_PCIE = 1
// CC_ACCEPTING_CLIENT_REQUESTS_FALSE as defined in nvml/nvml.h
CC_ACCEPTING_CLIENT_REQUESTS_FALSE = 0
// CC_ACCEPTING_CLIENT_REQUESTS_TRUE as defined in nvml/nvml.h
CC_ACCEPTING_CLIENT_REQUESTS_TRUE = 1
// GPU_CERT_CHAIN_SIZE as defined in nvml/nvml.h
GPU_CERT_CHAIN_SIZE = 4096
// GPU_ATTESTATION_CERT_CHAIN_SIZE as defined in nvml/nvml.h
GPU_ATTESTATION_CERT_CHAIN_SIZE = 5120
// CC_GPU_CEC_NONCE_SIZE as defined in nvml/nvml.h
CC_GPU_CEC_NONCE_SIZE = 32
// CC_GPU_ATTESTATION_REPORT_SIZE as defined in nvml/nvml.h
CC_GPU_ATTESTATION_REPORT_SIZE = 8192
// CC_GPU_CEC_ATTESTATION_REPORT_SIZE as defined in nvml/nvml.h
CC_GPU_CEC_ATTESTATION_REPORT_SIZE = 4096
// CC_CEC_ATTESTATION_REPORT_NOT_PRESENT as defined in nvml/nvml.h
CC_CEC_ATTESTATION_REPORT_NOT_PRESENT = 0
// CC_CEC_ATTESTATION_REPORT_PRESENT as defined in nvml/nvml.h
CC_CEC_ATTESTATION_REPORT_PRESENT = 1
// CC_KEY_ROTATION_THRESHOLD_ATTACKER_ADVANTAGE_MIN as defined in nvml/nvml.h
CC_KEY_ROTATION_THRESHOLD_ATTACKER_ADVANTAGE_MIN = 50
// CC_KEY_ROTATION_THRESHOLD_ATTACKER_ADVANTAGE_MAX as defined in nvml/nvml.h
CC_KEY_ROTATION_THRESHOLD_ATTACKER_ADVANTAGE_MAX = 75
// GPU_FABRIC_UUID_LEN as defined in nvml/nvml.h
GPU_FABRIC_UUID_LEN = 16
// GPU_FABRIC_STATE_NOT_SUPPORTED as defined in nvml/nvml.h
@ -562,6 +694,22 @@ const (
GPU_FABRIC_STATE_IN_PROGRESS = 2
// GPU_FABRIC_STATE_COMPLETED as defined in nvml/nvml.h
GPU_FABRIC_STATE_COMPLETED = 3
// GPU_FABRIC_HEALTH_MASK_DEGRADED_BW_NOT_SUPPORTED as defined in nvml/nvml.h
GPU_FABRIC_HEALTH_MASK_DEGRADED_BW_NOT_SUPPORTED = 0
// GPU_FABRIC_HEALTH_MASK_DEGRADED_BW_TRUE as defined in nvml/nvml.h
GPU_FABRIC_HEALTH_MASK_DEGRADED_BW_TRUE = 1
// GPU_FABRIC_HEALTH_MASK_DEGRADED_BW_FALSE as defined in nvml/nvml.h
GPU_FABRIC_HEALTH_MASK_DEGRADED_BW_FALSE = 2
// GPU_FABRIC_HEALTH_MASK_SHIFT_DEGRADED_BW as defined in nvml/nvml.h
GPU_FABRIC_HEALTH_MASK_SHIFT_DEGRADED_BW = 0
// GPU_FABRIC_HEALTH_MASK_WIDTH_DEGRADED_BW as defined in nvml/nvml.h
GPU_FABRIC_HEALTH_MASK_WIDTH_DEGRADED_BW = 17
// POWER_SCOPE_GPU as defined in nvml/nvml.h
POWER_SCOPE_GPU = 0
// POWER_SCOPE_MODULE as defined in nvml/nvml.h
POWER_SCOPE_MODULE = 1
// POWER_SCOPE_MEMORY as defined in nvml/nvml.h
POWER_SCOPE_MEMORY = 2
// INIT_FLAG_NO_GPUS as defined in nvml/nvml.h
INIT_FLAG_NO_GPUS = 1
// INIT_FLAG_NO_ATTACH as defined in nvml/nvml.h
@ -616,6 +764,8 @@ const (
GPU_INSTANCE_PROFILE_1_SLICE_REV2 = 9
// GPU_INSTANCE_PROFILE_COUNT as defined in nvml/nvml.h
GPU_INSTANCE_PROFILE_COUNT = 10
// GPU_INTSTANCE_PROFILE_CAPS_P2P as defined in nvml/nvml.h
GPU_INTSTANCE_PROFILE_CAPS_P2P = 1
// COMPUTE_INSTANCE_PROFILE_1_SLICE as defined in nvml/nvml.h
COMPUTE_INSTANCE_PROFILE_1_SLICE = 0
// COMPUTE_INSTANCE_PROFILE_2_SLICE as defined in nvml/nvml.h
@ -642,10 +792,6 @@ const (
GPM_METRICS_GET_VERSION = 1
// GPM_SUPPORT_VERSION as defined in nvml/nvml.h
GPM_SUPPORT_VERSION = 1
// COUNTER_COLLECTION_UNIT_STREAM_STATE_DISABLE as defined in nvml/nvml.h
COUNTER_COLLECTION_UNIT_STREAM_STATE_DISABLE = 0
// COUNTER_COLLECTION_UNIT_STREAM_STATE_ENABLE as defined in nvml/nvml.h
COUNTER_COLLECTION_UNIT_STREAM_STATE_ENABLE = 1
// NVLINK_POWER_STATE_HIGH_SPEED as defined in nvml/nvml.h
NVLINK_POWER_STATE_HIGH_SPEED = 0
// NVLINK_POWER_STATE_LOW as defined in nvml/nvml.h
@ -753,6 +899,7 @@ type GpuP2PStatus int32
const (
P2P_STATUS_OK GpuP2PStatus = iota
P2P_STATUS_CHIPSET_NOT_SUPPORED GpuP2PStatus = 1
P2P_STATUS_CHIPSET_NOT_SUPPORTED GpuP2PStatus = 1
P2P_STATUS_GPU_NOT_SUPPORTED GpuP2PStatus = 2
P2P_STATUS_IOH_TOPOLOGY_NOT_SUPPORTED GpuP2PStatus = 3
P2P_STATUS_DISABLED_BY_REGKEY GpuP2PStatus = 4
@ -769,6 +916,7 @@ const (
P2P_CAPS_INDEX_WRITE GpuP2PCapsIndex = 1
P2P_CAPS_INDEX_NVLINK GpuP2PCapsIndex = 2
P2P_CAPS_INDEX_ATOMICS GpuP2PCapsIndex = 3
P2P_CAPS_INDEX_PCI GpuP2PCapsIndex = 4
P2P_CAPS_INDEX_PROP GpuP2PCapsIndex = 4
P2P_CAPS_INDEX_UNKNOWN GpuP2PCapsIndex = 5
)
@ -785,7 +933,10 @@ const (
DEC_UTILIZATION_SAMPLES SamplingType = 4
PROCESSOR_CLK_SAMPLES SamplingType = 5
MEMORY_CLK_SAMPLES SamplingType = 6
SAMPLINGTYPE_COUNT SamplingType = 7
MODULE_POWER_SAMPLES SamplingType = 7
JPG_UTILIZATION_SAMPLES SamplingType = 8
OFA_UTILIZATION_SAMPLES SamplingType = 9
SAMPLINGTYPE_COUNT SamplingType = 10
)
// PcieUtilCounter as declared in nvml/nvml.h
@ -808,7 +959,8 @@ const (
VALUE_TYPE_UNSIGNED_LONG ValueType = 2
VALUE_TYPE_UNSIGNED_LONG_LONG ValueType = 3
VALUE_TYPE_SIGNED_LONG_LONG ValueType = 4
VALUE_TYPE_COUNT ValueType = 5
VALUE_TYPE_SIGNED_INT ValueType = 5
VALUE_TYPE_COUNT ValueType = 6
)
// PerfPolicyType as declared in nvml/nvml.h
@ -1028,6 +1180,9 @@ const (
ERROR_FREQ_NOT_SUPPORTED Return = 24
ERROR_ARGUMENT_VERSION_MISMATCH Return = 25
ERROR_DEPRECATED Return = 26
ERROR_NOT_READY Return = 27
ERROR_GPU_NOT_FOUND Return = 28
ERROR_INVALID_STATE Return = 29
ERROR_UNKNOWN Return = 999
)
@ -1137,7 +1292,12 @@ const (
DEVICE_VGPU_CAP_FRACTIONAL_MULTI_VGPU DeviceVgpuCapability = iota
DEVICE_VGPU_CAP_HETEROGENEOUS_TIMESLICE_PROFILES DeviceVgpuCapability = 1
DEVICE_VGPU_CAP_HETEROGENEOUS_TIMESLICE_SIZES DeviceVgpuCapability = 2
DEVICE_VGPU_CAP_COUNT DeviceVgpuCapability = 3
DEVICE_VGPU_CAP_READ_DEVICE_BUFFER_BW DeviceVgpuCapability = 3
DEVICE_VGPU_CAP_WRITE_DEVICE_BUFFER_BW DeviceVgpuCapability = 4
DEVICE_VGPU_CAP_DEVICE_STREAMING DeviceVgpuCapability = 5
DEVICE_VGPU_CAP_MINI_QUARTER_GPU DeviceVgpuCapability = 6
DEVICE_VGPU_CAP_COMPUTE_MEDIA_ENGINE_GPU DeviceVgpuCapability = 7
DEVICE_VGPU_CAP_COUNT DeviceVgpuCapability = 8
)
// GpuUtilizationDomainId as declared in nvml/nvml.h
@ -1174,8 +1334,10 @@ type EncoderType int32
// EncoderType enumeration from nvml/nvml.h
const (
ENCODER_QUERY_H264 EncoderType = iota
ENCODER_QUERY_HEVC EncoderType = 1
ENCODER_QUERY_H264 EncoderType = iota
ENCODER_QUERY_HEVC EncoderType = 1
ENCODER_QUERY_AV1 EncoderType = 2
ENCODER_QUERY_UNKNOWN EncoderType = 255
)
// FBCSessionType as declared in nvml/nvml.h

View File

@ -2085,28 +2085,28 @@ func (device nvmlDevice) GetGpuInstanceProfileInfo(profile int) (GpuInstanceProf
}
// nvml.DeviceGetGpuInstanceProfileInfoV()
type GpuInstanceProfileInfoV struct {
type GpuInstanceProfileInfoHandler struct {
device nvmlDevice
profile int
}
func (infoV GpuInstanceProfileInfoV) V1() (GpuInstanceProfileInfo, Return) {
return DeviceGetGpuInstanceProfileInfo(infoV.device, infoV.profile)
func (handler GpuInstanceProfileInfoHandler) V1() (GpuInstanceProfileInfo, Return) {
return DeviceGetGpuInstanceProfileInfo(handler.device, handler.profile)
}
func (infoV GpuInstanceProfileInfoV) V2() (GpuInstanceProfileInfo_v2, Return) {
func (handler GpuInstanceProfileInfoHandler) V2() (GpuInstanceProfileInfo_v2, Return) {
var info GpuInstanceProfileInfo_v2
info.Version = STRUCT_VERSION(info, 2)
ret := nvmlDeviceGetGpuInstanceProfileInfoV(infoV.device, uint32(infoV.profile), &info)
ret := nvmlDeviceGetGpuInstanceProfileInfoV(handler.device, uint32(handler.profile), &info)
return info, ret
}
func (l *library) DeviceGetGpuInstanceProfileInfoV(device Device, profile int) GpuInstanceProfileInfoV {
func (l *library) DeviceGetGpuInstanceProfileInfoV(device Device, profile int) GpuInstanceProfileInfoHandler {
return device.GetGpuInstanceProfileInfoV(profile)
}
func (device nvmlDevice) GetGpuInstanceProfileInfoV(profile int) GpuInstanceProfileInfoV {
return GpuInstanceProfileInfoV{device, profile}
func (device nvmlDevice) GetGpuInstanceProfileInfoV(profile int) GpuInstanceProfileInfoHandler {
return GpuInstanceProfileInfoHandler{device, profile}
}
// nvml.DeviceGetGpuInstancePossiblePlacements()
@ -2231,29 +2231,29 @@ func (gpuInstance nvmlGpuInstance) GetComputeInstanceProfileInfo(profile int, en
}
// nvml.GpuInstanceGetComputeInstanceProfileInfoV()
type ComputeInstanceProfileInfoV struct {
type ComputeInstanceProfileInfoHandler struct {
gpuInstance nvmlGpuInstance
profile int
engProfile int
}
func (infoV ComputeInstanceProfileInfoV) V1() (ComputeInstanceProfileInfo, Return) {
return GpuInstanceGetComputeInstanceProfileInfo(infoV.gpuInstance, infoV.profile, infoV.engProfile)
func (handler ComputeInstanceProfileInfoHandler) V1() (ComputeInstanceProfileInfo, Return) {
return GpuInstanceGetComputeInstanceProfileInfo(handler.gpuInstance, handler.profile, handler.engProfile)
}
func (infoV ComputeInstanceProfileInfoV) V2() (ComputeInstanceProfileInfo_v2, Return) {
func (handler ComputeInstanceProfileInfoHandler) V2() (ComputeInstanceProfileInfo_v2, Return) {
var info ComputeInstanceProfileInfo_v2
info.Version = STRUCT_VERSION(info, 2)
ret := nvmlGpuInstanceGetComputeInstanceProfileInfoV(infoV.gpuInstance, uint32(infoV.profile), uint32(infoV.engProfile), &info)
ret := nvmlGpuInstanceGetComputeInstanceProfileInfoV(handler.gpuInstance, uint32(handler.profile), uint32(handler.engProfile), &info)
return info, ret
}
func (l *library) GpuInstanceGetComputeInstanceProfileInfoV(gpuInstance GpuInstance, profile int, engProfile int) ComputeInstanceProfileInfoV {
func (l *library) GpuInstanceGetComputeInstanceProfileInfoV(gpuInstance GpuInstance, profile int, engProfile int) ComputeInstanceProfileInfoHandler {
return gpuInstance.GetComputeInstanceProfileInfoV(profile, engProfile)
}
func (gpuInstance nvmlGpuInstance) GetComputeInstanceProfileInfoV(profile int, engProfile int) ComputeInstanceProfileInfoV {
return ComputeInstanceProfileInfoV{gpuInstance, profile, engProfile}
func (gpuInstance nvmlGpuInstance) GetComputeInstanceProfileInfoV(profile int, engProfile int) ComputeInstanceProfileInfoHandler {
return ComputeInstanceProfileInfoHandler{gpuInstance, profile, engProfile}
}
// nvml.GpuInstanceGetComputeInstanceRemainingCapacity()
@ -2737,26 +2737,6 @@ func (device nvmlDevice) GetGpuFabricInfo() (GpuFabricInfo, Return) {
return gpuFabricInfo, ret
}
// nvml.DeviceCcuGetStreamState()
func (l *library) DeviceCcuGetStreamState(device Device) (int, Return) {
return device.CcuGetStreamState()
}
func (device nvmlDevice) CcuGetStreamState() (int, Return) {
var state uint32
ret := nvmlDeviceCcuGetStreamState(device, &state)
return int(state), ret
}
// nvml.DeviceCcuSetStreamState()
func (l *library) DeviceCcuSetStreamState(device Device, state int) Return {
return device.CcuSetStreamState(state)
}
func (device nvmlDevice) CcuSetStreamState(state int) Return {
return nvmlDeviceCcuSetStreamState(device, uint32(state))
}
// nvml.DeviceSetNvLinkDeviceLowPowerThreshold()
func (l *library) DeviceSetNvLinkDeviceLowPowerThreshold(device Device, info *NvLinkPowerThres) Return {
return device.SetNvLinkDeviceLowPowerThreshold(info)
@ -2765,3 +2745,313 @@ func (l *library) DeviceSetNvLinkDeviceLowPowerThreshold(device Device, info *Nv
func (device nvmlDevice) SetNvLinkDeviceLowPowerThreshold(info *NvLinkPowerThres) Return {
return nvmlDeviceSetNvLinkDeviceLowPowerThreshold(device, info)
}
// nvml.DeviceGetModuleId()
func (l *library) DeviceGetModuleId(device Device) (int, Return) {
return device.GetModuleId()
}
func (device nvmlDevice) GetModuleId() (int, Return) {
var moduleID uint32
ret := nvmlDeviceGetModuleId(device, &moduleID)
return int(moduleID), ret
}
// nvml.DeviceGetCurrentClocksEventReasons()
func (l *library) DeviceGetCurrentClocksEventReasons(device Device) (uint64, Return) {
return device.GetCurrentClocksEventReasons()
}
func (device nvmlDevice) GetCurrentClocksEventReasons() (uint64, Return) {
var clocksEventReasons uint64
ret := nvmlDeviceGetCurrentClocksEventReasons(device, &clocksEventReasons)
return clocksEventReasons, ret
}
// nvml.DeviceGetSupportedClocksEventReasons()
func (l *library) DeviceGetSupportedClocksEventReasons(device Device) (uint64, Return) {
return device.GetSupportedClocksEventReasons()
}
func (device nvmlDevice) GetSupportedClocksEventReasons() (uint64, Return) {
var supportedClocksEventReasons uint64
ret := nvmlDeviceGetSupportedClocksEventReasons(device, &supportedClocksEventReasons)
return supportedClocksEventReasons, ret
}
// nvml.DeviceGetJpgUtilization()
func (l *library) DeviceGetJpgUtilization(device Device) (uint32, uint32, Return) {
return device.GetJpgUtilization()
}
func (device nvmlDevice) GetJpgUtilization() (uint32, uint32, Return) {
var utilization, samplingPeriodUs uint32
ret := nvmlDeviceGetJpgUtilization(device, &utilization, &samplingPeriodUs)
return utilization, samplingPeriodUs, ret
}
// nvml.DeviceGetOfaUtilization()
func (l *library) DeviceGetOfaUtilization(device Device) (uint32, uint32, Return) {
return device.GetOfaUtilization()
}
func (device nvmlDevice) GetOfaUtilization() (uint32, uint32, Return) {
var utilization, samplingPeriodUs uint32
ret := nvmlDeviceGetOfaUtilization(device, &utilization, &samplingPeriodUs)
return utilization, samplingPeriodUs, ret
}
// nvml.DeviceGetRunningProcessDetailList()
func (l *library) DeviceGetRunningProcessDetailList(device Device) (ProcessDetailList, Return) {
return device.GetRunningProcessDetailList()
}
func (device nvmlDevice) GetRunningProcessDetailList() (ProcessDetailList, Return) {
var plist ProcessDetailList
ret := nvmlDeviceGetRunningProcessDetailList(device, &plist)
return plist, ret
}
// nvml.DeviceGetConfComputeMemSizeInfo()
func (l *library) DeviceGetConfComputeMemSizeInfo(device Device) (ConfComputeMemSizeInfo, Return) {
return device.GetConfComputeMemSizeInfo()
}
func (device nvmlDevice) GetConfComputeMemSizeInfo() (ConfComputeMemSizeInfo, Return) {
var memInfo ConfComputeMemSizeInfo
ret := nvmlDeviceGetConfComputeMemSizeInfo(device, &memInfo)
return memInfo, ret
}
// nvml.DeviceGetConfComputeProtectedMemoryUsage()
func (l *library) DeviceGetConfComputeProtectedMemoryUsage(device Device) (Memory, Return) {
return device.GetConfComputeProtectedMemoryUsage()
}
func (device nvmlDevice) GetConfComputeProtectedMemoryUsage() (Memory, Return) {
var memory Memory
ret := nvmlDeviceGetConfComputeProtectedMemoryUsage(device, &memory)
return memory, ret
}
// nvml.DeviceGetConfComputeGpuCertificate()
func (l *library) DeviceGetConfComputeGpuCertificate(device Device) (ConfComputeGpuCertificate, Return) {
return device.GetConfComputeGpuCertificate()
}
func (device nvmlDevice) GetConfComputeGpuCertificate() (ConfComputeGpuCertificate, Return) {
var gpuCert ConfComputeGpuCertificate
ret := nvmlDeviceGetConfComputeGpuCertificate(device, &gpuCert)
return gpuCert, ret
}
// nvml.DeviceGetConfComputeGpuAttestationReport()
func (l *library) DeviceGetConfComputeGpuAttestationReport(device Device) (ConfComputeGpuAttestationReport, Return) {
return device.GetConfComputeGpuAttestationReport()
}
func (device nvmlDevice) GetConfComputeGpuAttestationReport() (ConfComputeGpuAttestationReport, Return) {
var gpuAtstReport ConfComputeGpuAttestationReport
ret := nvmlDeviceGetConfComputeGpuAttestationReport(device, &gpuAtstReport)
return gpuAtstReport, ret
}
// nvml.DeviceSetConfComputeUnprotectedMemSize()
func (l *library) DeviceSetConfComputeUnprotectedMemSize(device Device, sizeKiB uint64) Return {
return device.SetConfComputeUnprotectedMemSize(sizeKiB)
}
func (device nvmlDevice) SetConfComputeUnprotectedMemSize(sizeKiB uint64) Return {
return nvmlDeviceSetConfComputeUnprotectedMemSize(device, sizeKiB)
}
// nvml.DeviceSetPowerManagementLimit_v2()
func (l *library) DeviceSetPowerManagementLimit_v2(device Device, powerValue *PowerValue_v2) Return {
return device.SetPowerManagementLimit_v2(powerValue)
}
func (device nvmlDevice) SetPowerManagementLimit_v2(powerValue *PowerValue_v2) Return {
return nvmlDeviceSetPowerManagementLimit_v2(device, powerValue)
}
// nvml.DeviceGetC2cModeInfoV()
type C2cModeInfoHandler struct {
device nvmlDevice
}
func (handler C2cModeInfoHandler) V1() (C2cModeInfo_v1, Return) {
var c2cModeInfo C2cModeInfo_v1
ret := nvmlDeviceGetC2cModeInfoV(handler.device, &c2cModeInfo)
return c2cModeInfo, ret
}
func (l *library) DeviceGetC2cModeInfoV(device Device) C2cModeInfoHandler {
return device.GetC2cModeInfoV()
}
func (device nvmlDevice) GetC2cModeInfoV() C2cModeInfoHandler {
return C2cModeInfoHandler{device}
}
// nvml.DeviceGetLastBBXFlushTime()
func (l *library) DeviceGetLastBBXFlushTime(device Device) (uint64, uint, Return) {
return device.GetLastBBXFlushTime()
}
func (device nvmlDevice) GetLastBBXFlushTime() (uint64, uint, Return) {
var timestamp uint64
var durationUs uint
ret := nvmlDeviceGetLastBBXFlushTime(device, &timestamp, &durationUs)
return timestamp, durationUs, ret
}
// nvml.DeviceGetNumaNodeId()
func (l *library) DeviceGetNumaNodeId(device Device) (int, Return) {
return device.GetNumaNodeId()
}
func (device nvmlDevice) GetNumaNodeId() (int, Return) {
var node uint32
ret := nvmlDeviceGetNumaNodeId(device, &node)
return int(node), ret
}
// nvml.DeviceGetPciInfoExt()
func (l *library) DeviceGetPciInfoExt(device Device) (PciInfoExt, Return) {
return device.GetPciInfoExt()
}
func (device nvmlDevice) GetPciInfoExt() (PciInfoExt, Return) {
var pciInfo PciInfoExt
ret := nvmlDeviceGetPciInfoExt(device, &pciInfo)
return pciInfo, ret
}
// nvml.DeviceGetGpuFabricInfoV()
type GpuFabricInfoHandler struct {
device nvmlDevice
}
func (handler GpuFabricInfoHandler) V1() (GpuFabricInfo, Return) {
return handler.device.GetGpuFabricInfo()
}
func (handler GpuFabricInfoHandler) V2() (GpuFabricInfo_v2, Return) {
var info GpuFabricInfoV
info.Version = STRUCT_VERSION(info, 2)
ret := nvmlDeviceGetGpuFabricInfoV(handler.device, &info)
return GpuFabricInfo_v2(info), ret
}
func (l *library) DeviceGetGpuFabricInfoV(device Device) GpuFabricInfoHandler {
return device.GetGpuFabricInfoV()
}
func (device nvmlDevice) GetGpuFabricInfoV() GpuFabricInfoHandler {
return GpuFabricInfoHandler{device}
}
// nvml.DeviceGetProcessesUtilizationInfo()
func (l *library) DeviceGetProcessesUtilizationInfo(device Device) (ProcessesUtilizationInfo, Return) {
return device.GetProcessesUtilizationInfo()
}
func (device nvmlDevice) GetProcessesUtilizationInfo() (ProcessesUtilizationInfo, Return) {
var processesUtilInfo ProcessesUtilizationInfo
ret := nvmlDeviceGetProcessesUtilizationInfo(device, &processesUtilInfo)
return processesUtilInfo, ret
}
// nvml.DeviceGetVgpuHeterogeneousMode()
func (l *library) DeviceGetVgpuHeterogeneousMode(device Device) (VgpuHeterogeneousMode, Return) {
return device.GetVgpuHeterogeneousMode()
}
func (device nvmlDevice) GetVgpuHeterogeneousMode() (VgpuHeterogeneousMode, Return) {
var heterogeneousMode VgpuHeterogeneousMode
ret := nvmlDeviceGetVgpuHeterogeneousMode(device, &heterogeneousMode)
return heterogeneousMode, ret
}
// nvml.DeviceSetVgpuHeterogeneousMode()
func (l *library) DeviceSetVgpuHeterogeneousMode(device Device, heterogeneousMode VgpuHeterogeneousMode) Return {
return device.SetVgpuHeterogeneousMode(heterogeneousMode)
}
func (device nvmlDevice) SetVgpuHeterogeneousMode(heterogeneousMode VgpuHeterogeneousMode) Return {
ret := nvmlDeviceSetVgpuHeterogeneousMode(device, &heterogeneousMode)
return ret
}
// nvml.DeviceGetVgpuTypeSupportedPlacements()
func (l *library) DeviceGetVgpuTypeSupportedPlacements(device Device, vgpuTypeId VgpuTypeId) (VgpuPlacementList, Return) {
return device.GetVgpuTypeSupportedPlacements(vgpuTypeId)
}
func (device nvmlDevice) GetVgpuTypeSupportedPlacements(vgpuTypeId VgpuTypeId) (VgpuPlacementList, Return) {
return vgpuTypeId.GetSupportedPlacements(device)
}
func (vgpuTypeId nvmlVgpuTypeId) GetSupportedPlacements(device Device) (VgpuPlacementList, Return) {
var placementList VgpuPlacementList
ret := nvmlDeviceGetVgpuTypeSupportedPlacements(nvmlDeviceHandle(device), vgpuTypeId, &placementList)
return placementList, ret
}
// nvml.DeviceGetVgpuTypeCreatablePlacements()
func (l *library) DeviceGetVgpuTypeCreatablePlacements(device Device, vgpuTypeId VgpuTypeId) (VgpuPlacementList, Return) {
return device.GetVgpuTypeCreatablePlacements(vgpuTypeId)
}
func (device nvmlDevice) GetVgpuTypeCreatablePlacements(vgpuTypeId VgpuTypeId) (VgpuPlacementList, Return) {
return vgpuTypeId.GetCreatablePlacements(device)
}
func (vgpuTypeId nvmlVgpuTypeId) GetCreatablePlacements(device Device) (VgpuPlacementList, Return) {
var placementList VgpuPlacementList
ret := nvmlDeviceGetVgpuTypeCreatablePlacements(nvmlDeviceHandle(device), vgpuTypeId, &placementList)
return placementList, ret
}
// nvml.DeviceSetVgpuCapabilities()
func (l *library) DeviceSetVgpuCapabilities(device Device, capability DeviceVgpuCapability, state EnableState) Return {
return device.SetVgpuCapabilities(capability, state)
}
func (device nvmlDevice) SetVgpuCapabilities(capability DeviceVgpuCapability, state EnableState) Return {
ret := nvmlDeviceSetVgpuCapabilities(device, capability, state)
return ret
}
// nvml.DeviceGetVgpuInstancesUtilizationInfo()
func (l *library) DeviceGetVgpuInstancesUtilizationInfo(device Device) (VgpuInstancesUtilizationInfo, Return) {
return device.GetVgpuInstancesUtilizationInfo()
}
func (device nvmlDevice) GetVgpuInstancesUtilizationInfo() (VgpuInstancesUtilizationInfo, Return) {
var vgpuUtilInfo VgpuInstancesUtilizationInfo
ret := nvmlDeviceGetVgpuInstancesUtilizationInfo(device, &vgpuUtilInfo)
return vgpuUtilInfo, ret
}
// nvml.DeviceGetVgpuProcessesUtilizationInfo()
func (l *library) DeviceGetVgpuProcessesUtilizationInfo(device Device) (VgpuProcessesUtilizationInfo, Return) {
return device.GetVgpuProcessesUtilizationInfo()
}
func (device nvmlDevice) GetVgpuProcessesUtilizationInfo() (VgpuProcessesUtilizationInfo, Return) {
var vgpuProcUtilInfo VgpuProcessesUtilizationInfo
ret := nvmlDeviceGetVgpuProcessesUtilizationInfo(device, &vgpuProcUtilInfo)
return vgpuProcUtilInfo, ret
}
// nvml.DeviceGetSramEccErrorStatus()
func (l *library) DeviceGetSramEccErrorStatus(device Device) (EccSramErrorStatus, Return) {
return device.GetSramEccErrorStatus()
}
func (device nvmlDevice) GetSramEccErrorStatus() (EccSramErrorStatus, Return) {
var status EccSramErrorStatus
ret := nvmlDeviceGetSramEccErrorStatus(device, &status)
return status, ret
}

View File

@ -51,20 +51,31 @@ func (g *nvmlGpmMetricsGetType) convert() *GpmMetricsGetType {
// nvml.GpmMetricsGet()
type GpmMetricsGetVType struct {
metricsGet *nvmlGpmMetricsGetType
metricsGet *GpmMetricsGetType
}
func (l *library) GpmMetricsGetV(metricsGet *GpmMetricsGetType) GpmMetricsGetVType {
return GpmMetricsGetVType{metricsGet.convert()}
return GpmMetricsGetVType{metricsGet}
}
// nvmlGpmMetricsGetStub is a stub function that can be overridden for testing.
var nvmlGpmMetricsGetStub = nvmlGpmMetricsGet
func (metricsGetV GpmMetricsGetVType) V1() Return {
metricsGetV.metricsGet.Version = 1
return nvmlGpmMetricsGet(metricsGetV.metricsGet)
return gpmMetricsGet(metricsGetV.metricsGet)
}
func (l *library) GpmMetricsGet(metricsGet *GpmMetricsGetType) Return {
metricsGet.Version = GPM_METRICS_GET_VERSION
return nvmlGpmMetricsGet(metricsGet.convert())
return gpmMetricsGet(metricsGet)
}
func gpmMetricsGet(metricsGet *GpmMetricsGetType) Return {
nvmlMetricsGet := metricsGet.convert()
ret := nvmlGpmMetricsGetStub(nvmlMetricsGet)
*metricsGet = *nvmlMetricsGet.convert()
return ret
}
// nvml.GpmSampleFree()
@ -139,3 +150,23 @@ func (device nvmlDevice) GpmMigSampleGet(gpuInstanceId int, gpmSample GpmSample)
func (gpmSample nvmlGpmSample) MigGet(device Device, gpuInstanceId int) Return {
return nvmlGpmMigSampleGet(nvmlDeviceHandle(device), uint32(gpuInstanceId), gpmSample)
}
// nvml.GpmQueryIfStreamingEnabled()
func (l *library) GpmQueryIfStreamingEnabled(device Device) (uint32, Return) {
return device.GpmQueryIfStreamingEnabled()
}
func (device nvmlDevice) GpmQueryIfStreamingEnabled() (uint32, Return) {
var state uint32
ret := nvmlGpmQueryIfStreamingEnabled(device, &state)
return state, ret
}
// nvml.GpmSetStreamingEnabled()
func (l *library) GpmSetStreamingEnabled(device Device, state uint32) Return {
return device.GpmSetStreamingEnabled(state)
}
func (device nvmlDevice) GpmSetStreamingEnabled(state uint32) Return {
return nvmlGpmSetStreamingEnabled(device, state)
}

File diff suppressed because it is too large Load Diff

View File

@ -36,7 +36,7 @@ var _ nvml.GpuInstance = &GpuInstance{}
// GetComputeInstanceProfileInfoFunc: func(n1 int, n2 int) (nvml.ComputeInstanceProfileInfo, nvml.Return) {
// panic("mock out the GetComputeInstanceProfileInfo method")
// },
// GetComputeInstanceProfileInfoVFunc: func(n1 int, n2 int) nvml.ComputeInstanceProfileInfoV {
// GetComputeInstanceProfileInfoVFunc: func(n1 int, n2 int) nvml.ComputeInstanceProfileInfoHandler {
// panic("mock out the GetComputeInstanceProfileInfoV method")
// },
// GetComputeInstanceRemainingCapacityFunc: func(computeInstanceProfileInfo *nvml.ComputeInstanceProfileInfo) (int, nvml.Return) {
@ -74,7 +74,7 @@ type GpuInstance struct {
GetComputeInstanceProfileInfoFunc func(n1 int, n2 int) (nvml.ComputeInstanceProfileInfo, nvml.Return)
// GetComputeInstanceProfileInfoVFunc mocks the GetComputeInstanceProfileInfoV method.
GetComputeInstanceProfileInfoVFunc func(n1 int, n2 int) nvml.ComputeInstanceProfileInfoV
GetComputeInstanceProfileInfoVFunc func(n1 int, n2 int) nvml.ComputeInstanceProfileInfoHandler
// GetComputeInstanceRemainingCapacityFunc mocks the GetComputeInstanceRemainingCapacity method.
GetComputeInstanceRemainingCapacityFunc func(computeInstanceProfileInfo *nvml.ComputeInstanceProfileInfo) (int, nvml.Return)
@ -348,7 +348,7 @@ func (mock *GpuInstance) GetComputeInstanceProfileInfoCalls() []struct {
}
// GetComputeInstanceProfileInfoV calls GetComputeInstanceProfileInfoVFunc.
func (mock *GpuInstance) GetComputeInstanceProfileInfoV(n1 int, n2 int) nvml.ComputeInstanceProfileInfoV {
func (mock *GpuInstance) GetComputeInstanceProfileInfoV(n1 int, n2 int) nvml.ComputeInstanceProfileInfoHandler {
if mock.GetComputeInstanceProfileInfoVFunc == nil {
panic("GpuInstance.GetComputeInstanceProfileInfoVFunc: method is nil but GpuInstance.GetComputeInstanceProfileInfoV was just called")
}

File diff suppressed because it is too large Load Diff

View File

@ -24,6 +24,9 @@ var _ nvml.VgpuTypeId = &VgpuTypeId{}
// GetClassFunc: func() (string, nvml.Return) {
// panic("mock out the GetClass method")
// },
// GetCreatablePlacementsFunc: func(device nvml.Device) (nvml.VgpuPlacementList, nvml.Return) {
// panic("mock out the GetCreatablePlacements method")
// },
// GetDeviceIDFunc: func() (uint64, uint64, nvml.Return) {
// panic("mock out the GetDeviceID method")
// },
@ -54,6 +57,9 @@ var _ nvml.VgpuTypeId = &VgpuTypeId{}
// GetResolutionFunc: func(n int) (uint32, uint32, nvml.Return) {
// panic("mock out the GetResolution method")
// },
// GetSupportedPlacementsFunc: func(device nvml.Device) (nvml.VgpuPlacementList, nvml.Return) {
// panic("mock out the GetSupportedPlacements method")
// },
// }
//
// // use mockedVgpuTypeId in code that requires nvml.VgpuTypeId
@ -67,6 +73,9 @@ type VgpuTypeId struct {
// GetClassFunc mocks the GetClass method.
GetClassFunc func() (string, nvml.Return)
// GetCreatablePlacementsFunc mocks the GetCreatablePlacements method.
GetCreatablePlacementsFunc func(device nvml.Device) (nvml.VgpuPlacementList, nvml.Return)
// GetDeviceIDFunc mocks the GetDeviceID method.
GetDeviceIDFunc func() (uint64, uint64, nvml.Return)
@ -97,6 +106,9 @@ type VgpuTypeId struct {
// GetResolutionFunc mocks the GetResolution method.
GetResolutionFunc func(n int) (uint32, uint32, nvml.Return)
// GetSupportedPlacementsFunc mocks the GetSupportedPlacements method.
GetSupportedPlacementsFunc func(device nvml.Device) (nvml.VgpuPlacementList, nvml.Return)
// calls tracks calls to the methods.
calls struct {
// GetCapabilities holds details about calls to the GetCapabilities method.
@ -107,6 +119,11 @@ type VgpuTypeId struct {
// GetClass holds details about calls to the GetClass method.
GetClass []struct {
}
// GetCreatablePlacements holds details about calls to the GetCreatablePlacements method.
GetCreatablePlacements []struct {
// Device is the device argument value.
Device nvml.Device
}
// GetDeviceID holds details about calls to the GetDeviceID method.
GetDeviceID []struct {
}
@ -141,9 +158,15 @@ type VgpuTypeId struct {
// N is the n argument value.
N int
}
// GetSupportedPlacements holds details about calls to the GetSupportedPlacements method.
GetSupportedPlacements []struct {
// Device is the device argument value.
Device nvml.Device
}
}
lockGetCapabilities sync.RWMutex
lockGetClass sync.RWMutex
lockGetCreatablePlacements sync.RWMutex
lockGetDeviceID sync.RWMutex
lockGetFrameRateLimit sync.RWMutex
lockGetFramebufferSize sync.RWMutex
@ -154,6 +177,7 @@ type VgpuTypeId struct {
lockGetName sync.RWMutex
lockGetNumDisplayHeads sync.RWMutex
lockGetResolution sync.RWMutex
lockGetSupportedPlacements sync.RWMutex
}
// GetCapabilities calls GetCapabilitiesFunc.
@ -215,6 +239,38 @@ func (mock *VgpuTypeId) GetClassCalls() []struct {
return calls
}
// GetCreatablePlacements calls GetCreatablePlacementsFunc.
func (mock *VgpuTypeId) GetCreatablePlacements(device nvml.Device) (nvml.VgpuPlacementList, nvml.Return) {
if mock.GetCreatablePlacementsFunc == nil {
panic("VgpuTypeId.GetCreatablePlacementsFunc: method is nil but VgpuTypeId.GetCreatablePlacements was just called")
}
callInfo := struct {
Device nvml.Device
}{
Device: device,
}
mock.lockGetCreatablePlacements.Lock()
mock.calls.GetCreatablePlacements = append(mock.calls.GetCreatablePlacements, callInfo)
mock.lockGetCreatablePlacements.Unlock()
return mock.GetCreatablePlacementsFunc(device)
}
// GetCreatablePlacementsCalls gets all the calls that were made to GetCreatablePlacements.
// Check the length with:
//
// len(mockedVgpuTypeId.GetCreatablePlacementsCalls())
func (mock *VgpuTypeId) GetCreatablePlacementsCalls() []struct {
Device nvml.Device
} {
var calls []struct {
Device nvml.Device
}
mock.lockGetCreatablePlacements.RLock()
calls = mock.calls.GetCreatablePlacements
mock.lockGetCreatablePlacements.RUnlock()
return calls
}
// GetDeviceID calls GetDeviceIDFunc.
func (mock *VgpuTypeId) GetDeviceID() (uint64, uint64, nvml.Return) {
if mock.GetDeviceIDFunc == nil {
@ -494,3 +550,35 @@ func (mock *VgpuTypeId) GetResolutionCalls() []struct {
mock.lockGetResolution.RUnlock()
return calls
}
// GetSupportedPlacements calls GetSupportedPlacementsFunc.
func (mock *VgpuTypeId) GetSupportedPlacements(device nvml.Device) (nvml.VgpuPlacementList, nvml.Return) {
if mock.GetSupportedPlacementsFunc == nil {
panic("VgpuTypeId.GetSupportedPlacementsFunc: method is nil but VgpuTypeId.GetSupportedPlacements was just called")
}
callInfo := struct {
Device nvml.Device
}{
Device: device,
}
mock.lockGetSupportedPlacements.Lock()
mock.calls.GetSupportedPlacements = append(mock.calls.GetSupportedPlacements, callInfo)
mock.lockGetSupportedPlacements.Unlock()
return mock.GetSupportedPlacementsFunc(device)
}
// GetSupportedPlacementsCalls gets all the calls that were made to GetSupportedPlacements.
// Check the length with:
//
// len(mockedVgpuTypeId.GetSupportedPlacementsCalls())
func (mock *VgpuTypeId) GetSupportedPlacementsCalls() []struct {
Device nvml.Device
} {
var calls []struct {
Device nvml.Device
}
mock.lockGetSupportedPlacements.RLock()
calls = mock.calls.GetSupportedPlacements
mock.lockGetSupportedPlacements.RUnlock()
return calls
}

File diff suppressed because it is too large Load Diff

File diff suppressed because it is too large Load Diff

View File

@ -79,3 +79,60 @@ func (l *library) SystemGetTopologyGpuSet(cpuNumber int) ([]Device, Return) {
ret = nvmlSystemGetTopologyGpuSet(uint32(cpuNumber), &count, &deviceArray[0])
return convertSlice[nvmlDevice, Device](deviceArray), ret
}
// nvml.SystemGetConfComputeCapabilities()
func (l *library) SystemGetConfComputeCapabilities() (ConfComputeSystemCaps, Return) {
var capabilities ConfComputeSystemCaps
ret := nvmlSystemGetConfComputeCapabilities(&capabilities)
return capabilities, ret
}
// nvml.SystemGetConfComputeState()
func SystemGetConfComputeState() (ConfComputeSystemState, Return) {
var state ConfComputeSystemState
ret := nvmlSystemGetConfComputeState(&state)
return state, ret
}
// nvml.SystemGetConfComputeGpusReadyState()
func SystemGetConfComputeGpusReadyState() (uint32, Return) {
var isAcceptingWork uint32
ret := nvmlSystemGetConfComputeGpusReadyState(&isAcceptingWork)
return isAcceptingWork, ret
}
// nvml.SystemSetConfComputeGpusReadyState()
func SystemSetConfComputeGpusReadyState(isAcceptingWork uint32) Return {
return nvmlSystemSetConfComputeGpusReadyState(isAcceptingWork)
}
// nvml.SystemSetNvlinkBwMode()
func SystemSetNvlinkBwMode(nvlinkBwMode uint32) Return {
return nvmlSystemSetNvlinkBwMode(nvlinkBwMode)
}
// nvml.SystemGetNvlinkBwMode()
func SystemGetNvlinkBwMode() (uint32, Return) {
var nvlinkBwMode uint32
ret := nvmlSystemGetNvlinkBwMode(&nvlinkBwMode)
return nvlinkBwMode, ret
}
// nvml.SystemGetConfComputeKeyRotationThresholdInfo()
func (l *library) SystemGetConfComputeKeyRotationThresholdInfo() (ConfComputeGetKeyRotationThresholdInfo, Return) {
var keyRotationThresholdInfo ConfComputeGetKeyRotationThresholdInfo
ret := nvmlSystemGetConfComputeKeyRotationThresholdInfo(&keyRotationThresholdInfo)
return keyRotationThresholdInfo, ret
}
// nvml.SystemGetConfComputeSettings()
func (l *library) SystemGetConfComputeSettings() (SystemConfComputeSettings, Return) {
var settings SystemConfComputeSettings
ret := nvmlSystemGetConfComputeSettings(&settings)
return settings, ret
}
// nvml.SystemSetConfComputeKeyRotationThresholdInfo()
func (l *library) SystemSetConfComputeKeyRotationThresholdInfo(keyRotationThresholdInfo ConfComputeSetKeyRotationThresholdInfo) Return {
return nvmlSystemSetConfComputeKeyRotationThresholdInfo(&keyRotationThresholdInfo)
}

View File

@ -9,6 +9,30 @@ type nvmlDevice struct {
Handle *_Ctype_struct_nvmlDevice_st
}
type PciInfoExt_v1 struct {
Version uint32
Domain uint32
Bus uint32
Device uint32
PciDeviceId uint32
PciSubSystemId uint32
BaseClass uint32
SubClass uint32
BusId [32]int8
}
type PciInfoExt struct {
Version uint32
Domain uint32
Bus uint32
Device uint32
PciDeviceId uint32
PciSubSystemId uint32
BaseClass uint32
SubClass uint32
BusId [32]int8
}
type PciInfo struct {
BusIdLegacy [16]int8
Domain uint32
@ -70,6 +94,28 @@ type ProcessInfo struct {
ComputeInstanceId uint32
}
type ProcessDetail_v1 struct {
Pid uint32
UsedGpuMemory uint64
GpuInstanceId uint32
ComputeInstanceId uint32
UsedGpuCcProtectedMemory uint64
}
type ProcessDetailList_v1 struct {
Version uint32
Mode uint32
NumProcArrayEntries uint32
ProcArray *ProcessDetail_v1
}
type ProcessDetailList struct {
Version uint32
Mode uint32
NumProcArrayEntries uint32
ProcArray *ProcessDetail_v1
}
type DeviceAttributes struct {
MultiprocessorCount uint32
SharedCopyEngineCount uint32
@ -82,6 +128,10 @@ type DeviceAttributes struct {
MemorySizeMB uint64
}
type C2cModeInfo_v1 struct {
IsC2cEnabled uint32
}
type RowRemapperHistogramValues struct {
Max uint32
High uint32
@ -147,6 +197,40 @@ type nvmlVgpuTypeId uint32
type nvmlVgpuInstance uint32
type VgpuHeterogeneousMode_v1 struct {
Version uint32
Mode uint32
}
type VgpuHeterogeneousMode struct {
Version uint32
Mode uint32
}
type VgpuPlacementId_v1 struct {
Version uint32
PlacementId uint32
}
type VgpuPlacementId struct {
Version uint32
PlacementId uint32
}
type VgpuPlacementList_v1 struct {
Version uint32
PlacementSize uint32
Count uint32
PlacementIds *uint32
}
type VgpuPlacementList struct {
Version uint32
PlacementSize uint32
Count uint32
PlacementIds *uint32
}
type VgpuInstanceUtilizationSample struct {
VgpuInstance uint32
TimeStamp uint64
@ -156,6 +240,34 @@ type VgpuInstanceUtilizationSample struct {
DecUtil [8]byte
}
type VgpuInstanceUtilizationInfo_v1 struct {
TimeStamp uint64
VgpuInstance uint32
Pad_cgo_0 [4]byte
SmUtil [8]byte
MemUtil [8]byte
EncUtil [8]byte
DecUtil [8]byte
JpgUtil [8]byte
OfaUtil [8]byte
}
type VgpuInstancesUtilizationInfo_v1 struct {
Version uint32
SampleValType uint32
VgpuInstanceCount uint32
LastSeenTimeStamp uint64
VgpuUtilArray *VgpuInstanceUtilizationInfo_v1
}
type VgpuInstancesUtilizationInfo struct {
Version uint32
SampleValType uint32
VgpuInstanceCount uint32
LastSeenTimeStamp uint64
VgpuUtilArray *VgpuInstanceUtilizationInfo_v1
}
type VgpuProcessUtilizationSample struct {
VgpuInstance uint32
Pid uint32
@ -167,6 +279,33 @@ type VgpuProcessUtilizationSample struct {
DecUtil uint32
}
type VgpuProcessUtilizationInfo_v1 struct {
ProcessName [64]int8
TimeStamp uint64
VgpuInstance uint32
Pid uint32
SmUtil uint32
MemUtil uint32
EncUtil uint32
DecUtil uint32
JpgUtil uint32
OfaUtil uint32
}
type VgpuProcessesUtilizationInfo_v1 struct {
Version uint32
VgpuProcessCount uint32
LastSeenTimeStamp uint64
VgpuProcUtilArray *VgpuProcessUtilizationInfo_v1
}
type VgpuProcessesUtilizationInfo struct {
Version uint32
VgpuProcessCount uint32
LastSeenTimeStamp uint64
VgpuProcUtilArray *VgpuProcessUtilizationInfo_v1
}
type VgpuSchedulerParamsVgpuSchedDataWithARR struct {
AvgFactor uint32
Timeslice uint32
@ -192,7 +331,7 @@ type VgpuSchedulerLogEntry struct {
type VgpuSchedulerLog struct {
EngineId uint32
SchedulerPolicy uint32
IsEnabledARR uint32
ArrMode uint32
SchedulerParams [8]byte
EntriesCount uint32
LogEntries [200]VgpuSchedulerLogEntry
@ -200,7 +339,7 @@ type VgpuSchedulerLog struct {
type VgpuSchedulerGetState struct {
SchedulerPolicy uint32
IsEnabledARR uint32
ArrMode uint32
SchedulerParams [8]byte
}
@ -260,6 +399,32 @@ type ProcessUtilizationSample struct {
DecUtil uint32
}
type ProcessUtilizationInfo_v1 struct {
TimeStamp uint64
Pid uint32
SmUtil uint32
MemUtil uint32
EncUtil uint32
DecUtil uint32
JpgUtil uint32
OfaUtil uint32
Pad_cgo_0 [4]byte
}
type ProcessesUtilizationInfo_v1 struct {
Version uint32
ProcessSamplesCount uint32
LastSeenTimeStamp uint64
ProcUtilArray *ProcessUtilizationInfo_v1
}
type ProcessesUtilizationInfo struct {
Version uint32
ProcessSamplesCount uint32
LastSeenTimeStamp uint64
ProcUtilArray *ProcessUtilizationInfo_v1
}
type GridLicenseExpiry struct {
Year uint32
Month uint16
@ -286,6 +451,40 @@ type GridLicensableFeatures struct {
GridLicensableFeatures [3]GridLicensableFeature
}
type EccSramErrorStatus_v1 struct {
Version uint32
AggregateUncParity uint64
AggregateUncSecDed uint64
AggregateCor uint64
VolatileUncParity uint64
VolatileUncSecDed uint64
VolatileCor uint64
AggregateUncBucketL2 uint64
AggregateUncBucketSm uint64
AggregateUncBucketPcie uint64
AggregateUncBucketMcu uint64
AggregateUncBucketOther uint64
BThresholdExceeded uint32
Pad_cgo_0 [4]byte
}
type EccSramErrorStatus struct {
Version uint32
AggregateUncParity uint64
AggregateUncSecDed uint64
AggregateCor uint64
VolatileUncParity uint64
VolatileUncSecDed uint64
VolatileCor uint64
AggregateUncBucketL2 uint64
AggregateUncBucketSm uint64
AggregateUncBucketPcie uint64
AggregateUncBucketMcu uint64
AggregateUncBucketOther uint64
BThresholdExceeded uint32
Pad_cgo_0 [4]byte
}
type DeviceArchitecture uint32
type BusType uint32
@ -408,16 +607,110 @@ type FBCSessionInfo struct {
AverageLatency uint32
}
type ConfComputeSystemCaps struct {
CpuCaps uint32
GpusCaps uint32
}
type ConfComputeSystemState struct {
Environment uint32
CcFeature uint32
DevToolsMode uint32
}
type SystemConfComputeSettings_v1 struct {
Version uint32
Environment uint32
CcFeature uint32
DevToolsMode uint32
MultiGpuMode uint32
}
type SystemConfComputeSettings struct {
Version uint32
Environment uint32
CcFeature uint32
DevToolsMode uint32
MultiGpuMode uint32
}
type ConfComputeMemSizeInfo struct {
ProtectedMemSizeKib uint64
UnprotectedMemSizeKib uint64
}
type ConfComputeGpuCertificate struct {
CertChainSize uint32
AttestationCertChainSize uint32
CertChain [4096]uint8
AttestationCertChain [5120]uint8
}
type ConfComputeGpuAttestationReport struct {
IsCecAttestationReportPresent uint32
AttestationReportSize uint32
CecAttestationReportSize uint32
Nonce [32]uint8
AttestationReport [8192]uint8
CecAttestationReport [4096]uint8
}
type ConfComputeSetKeyRotationThresholdInfo_v1 struct {
Version uint32
MaxAttackerAdvantage uint64
}
type ConfComputeSetKeyRotationThresholdInfo struct {
Version uint32
MaxAttackerAdvantage uint64
}
type ConfComputeGetKeyRotationThresholdInfo_v1 struct {
Version uint32
AttackerAdvantage uint64
}
type ConfComputeGetKeyRotationThresholdInfo struct {
Version uint32
AttackerAdvantage uint64
}
type GpuFabricState byte
type GpuFabricInfo struct {
ClusterUuid [16]int8
ClusterUuid [16]uint8
Status uint32
PartitionId uint32
CliqueId uint32
State uint8
Pad_cgo_0 [3]byte
}
type GpuFabricInfo_v2 struct {
Version uint32
ClusterUuid [16]uint8
Status uint32
CliqueId uint32
State uint8
HealthMask uint32
}
type GpuFabricInfoV struct {
Version uint32
ClusterUuid [16]uint8
Status uint32
CliqueId uint32
State uint8
HealthMask uint32
}
type PowerScopeType byte
type PowerValue_v2 struct {
Version uint32
PowerScope uint8
PowerValueMw uint32
}
type AffinityScope uint32
type VgpuVersion struct {
@ -494,6 +787,23 @@ type GpuInstanceProfileInfo_v2 struct {
Name [96]int8
}
type GpuInstanceProfileInfo_v3 struct {
Version uint32
Id uint32
SliceCount uint32
InstanceCount uint32
MultiprocessorCount uint32
CopyEngineCount uint32
DecoderCount uint32
EncoderCount uint32
JpegCount uint32
OfaCount uint32
MemorySizeMB uint64
Name [96]int8
Capabilities uint32
Pad_cgo_0 [4]byte
}
type nvmlGpuInstanceInfo struct {
Device nvmlDevice
Id uint32
@ -536,6 +846,21 @@ type ComputeInstanceProfileInfo_v2 struct {
Name [96]int8
}
type ComputeInstanceProfileInfo_v3 struct {
Version uint32
Id uint32
SliceCount uint32
InstanceCount uint32
MultiprocessorCount uint32
SharedCopyEngineCount uint32
SharedDecoderCount uint32
SharedEncoderCount uint32
SharedJpegCount uint32
SharedOfaCount uint32
Name [96]int8
Capabilities uint32
}
type nvmlComputeInstanceInfo struct {
Device nvmlDevice
GpuInstance nvmlGpuInstance

View File

@ -22,8 +22,6 @@ package nvml
var (
ComputeInstanceDestroy = libnvml.ComputeInstanceDestroy
ComputeInstanceGetInfo = libnvml.ComputeInstanceGetInfo
DeviceCcuGetStreamState = libnvml.DeviceCcuGetStreamState
DeviceCcuSetStreamState = libnvml.DeviceCcuSetStreamState
DeviceClearAccountingPids = libnvml.DeviceClearAccountingPids
DeviceClearCpuAffinity = libnvml.DeviceClearCpuAffinity
DeviceClearEccErrorCounts = libnvml.DeviceClearEccErrorCounts
@ -49,12 +47,17 @@ var (
DeviceGetBrand = libnvml.DeviceGetBrand
DeviceGetBridgeChipInfo = libnvml.DeviceGetBridgeChipInfo
DeviceGetBusType = libnvml.DeviceGetBusType
DeviceGetC2cModeInfoV = libnvml.DeviceGetC2cModeInfoV
DeviceGetClkMonStatus = libnvml.DeviceGetClkMonStatus
DeviceGetClock = libnvml.DeviceGetClock
DeviceGetClockInfo = libnvml.DeviceGetClockInfo
DeviceGetComputeInstanceId = libnvml.DeviceGetComputeInstanceId
DeviceGetComputeMode = libnvml.DeviceGetComputeMode
DeviceGetComputeRunningProcesses = libnvml.DeviceGetComputeRunningProcesses
DeviceGetConfComputeGpuAttestationReport = libnvml.DeviceGetConfComputeGpuAttestationReport
DeviceGetConfComputeGpuCertificate = libnvml.DeviceGetConfComputeGpuCertificate
DeviceGetConfComputeMemSizeInfo = libnvml.DeviceGetConfComputeMemSizeInfo
DeviceGetConfComputeProtectedMemoryUsage = libnvml.DeviceGetConfComputeProtectedMemoryUsage
DeviceGetCount = libnvml.DeviceGetCount
DeviceGetCpuAffinity = libnvml.DeviceGetCpuAffinity
DeviceGetCpuAffinityWithinScope = libnvml.DeviceGetCpuAffinityWithinScope
@ -62,6 +65,7 @@ var (
DeviceGetCudaComputeCapability = libnvml.DeviceGetCudaComputeCapability
DeviceGetCurrPcieLinkGeneration = libnvml.DeviceGetCurrPcieLinkGeneration
DeviceGetCurrPcieLinkWidth = libnvml.DeviceGetCurrPcieLinkWidth
DeviceGetCurrentClocksEventReasons = libnvml.DeviceGetCurrentClocksEventReasons
DeviceGetCurrentClocksThrottleReasons = libnvml.DeviceGetCurrentClocksThrottleReasons
DeviceGetDecoderUtilization = libnvml.DeviceGetDecoderUtilization
DeviceGetDefaultApplicationsClock = libnvml.DeviceGetDefaultApplicationsClock
@ -87,6 +91,7 @@ var (
DeviceGetGpcClkMinMaxVfOffset = libnvml.DeviceGetGpcClkMinMaxVfOffset
DeviceGetGpcClkVfOffset = libnvml.DeviceGetGpcClkVfOffset
DeviceGetGpuFabricInfo = libnvml.DeviceGetGpuFabricInfo
DeviceGetGpuFabricInfoV = libnvml.DeviceGetGpuFabricInfoV
DeviceGetGpuInstanceById = libnvml.DeviceGetGpuInstanceById
DeviceGetGpuInstanceId = libnvml.DeviceGetGpuInstanceId
DeviceGetGpuInstancePossiblePlacements = libnvml.DeviceGetGpuInstancePossiblePlacements
@ -110,6 +115,8 @@ var (
DeviceGetInforomImageVersion = libnvml.DeviceGetInforomImageVersion
DeviceGetInforomVersion = libnvml.DeviceGetInforomVersion
DeviceGetIrqNum = libnvml.DeviceGetIrqNum
DeviceGetJpgUtilization = libnvml.DeviceGetJpgUtilization
DeviceGetLastBBXFlushTime = libnvml.DeviceGetLastBBXFlushTime
DeviceGetMPSComputeRunningProcesses = libnvml.DeviceGetMPSComputeRunningProcesses
DeviceGetMaxClockInfo = libnvml.DeviceGetMaxClockInfo
DeviceGetMaxCustomerBoostClock = libnvml.DeviceGetMaxCustomerBoostClock
@ -128,10 +135,12 @@ var (
DeviceGetMinMaxClockOfPState = libnvml.DeviceGetMinMaxClockOfPState
DeviceGetMinMaxFanSpeed = libnvml.DeviceGetMinMaxFanSpeed
DeviceGetMinorNumber = libnvml.DeviceGetMinorNumber
DeviceGetModuleId = libnvml.DeviceGetModuleId
DeviceGetMultiGpuBoard = libnvml.DeviceGetMultiGpuBoard
DeviceGetName = libnvml.DeviceGetName
DeviceGetNumFans = libnvml.DeviceGetNumFans
DeviceGetNumGpuCores = libnvml.DeviceGetNumGpuCores
DeviceGetNumaNodeId = libnvml.DeviceGetNumaNodeId
DeviceGetNvLinkCapability = libnvml.DeviceGetNvLinkCapability
DeviceGetNvLinkErrorCounter = libnvml.DeviceGetNvLinkErrorCounter
DeviceGetNvLinkRemoteDeviceType = libnvml.DeviceGetNvLinkRemoteDeviceType
@ -140,8 +149,10 @@ var (
DeviceGetNvLinkUtilizationControl = libnvml.DeviceGetNvLinkUtilizationControl
DeviceGetNvLinkUtilizationCounter = libnvml.DeviceGetNvLinkUtilizationCounter
DeviceGetNvLinkVersion = libnvml.DeviceGetNvLinkVersion
DeviceGetOfaUtilization = libnvml.DeviceGetOfaUtilization
DeviceGetP2PStatus = libnvml.DeviceGetP2PStatus
DeviceGetPciInfo = libnvml.DeviceGetPciInfo
DeviceGetPciInfoExt = libnvml.DeviceGetPciInfoExt
DeviceGetPcieLinkMaxSpeed = libnvml.DeviceGetPcieLinkMaxSpeed
DeviceGetPcieReplayCounter = libnvml.DeviceGetPcieReplayCounter
DeviceGetPcieSpeed = libnvml.DeviceGetPcieSpeed
@ -157,13 +168,17 @@ var (
DeviceGetPowerState = libnvml.DeviceGetPowerState
DeviceGetPowerUsage = libnvml.DeviceGetPowerUsage
DeviceGetProcessUtilization = libnvml.DeviceGetProcessUtilization
DeviceGetProcessesUtilizationInfo = libnvml.DeviceGetProcessesUtilizationInfo
DeviceGetRemappedRows = libnvml.DeviceGetRemappedRows
DeviceGetRetiredPages = libnvml.DeviceGetRetiredPages
DeviceGetRetiredPagesPendingStatus = libnvml.DeviceGetRetiredPagesPendingStatus
DeviceGetRetiredPages_v2 = libnvml.DeviceGetRetiredPages_v2
DeviceGetRowRemapperHistogram = libnvml.DeviceGetRowRemapperHistogram
DeviceGetRunningProcessDetailList = libnvml.DeviceGetRunningProcessDetailList
DeviceGetSamples = libnvml.DeviceGetSamples
DeviceGetSerial = libnvml.DeviceGetSerial
DeviceGetSramEccErrorStatus = libnvml.DeviceGetSramEccErrorStatus
DeviceGetSupportedClocksEventReasons = libnvml.DeviceGetSupportedClocksEventReasons
DeviceGetSupportedClocksThrottleReasons = libnvml.DeviceGetSupportedClocksThrottleReasons
DeviceGetSupportedEventTypes = libnvml.DeviceGetSupportedEventTypes
DeviceGetSupportedGraphicsClocks = libnvml.DeviceGetSupportedGraphicsClocks
@ -182,11 +197,16 @@ var (
DeviceGetUtilizationRates = libnvml.DeviceGetUtilizationRates
DeviceGetVbiosVersion = libnvml.DeviceGetVbiosVersion
DeviceGetVgpuCapabilities = libnvml.DeviceGetVgpuCapabilities
DeviceGetVgpuHeterogeneousMode = libnvml.DeviceGetVgpuHeterogeneousMode
DeviceGetVgpuInstancesUtilizationInfo = libnvml.DeviceGetVgpuInstancesUtilizationInfo
DeviceGetVgpuMetadata = libnvml.DeviceGetVgpuMetadata
DeviceGetVgpuProcessUtilization = libnvml.DeviceGetVgpuProcessUtilization
DeviceGetVgpuProcessesUtilizationInfo = libnvml.DeviceGetVgpuProcessesUtilizationInfo
DeviceGetVgpuSchedulerCapabilities = libnvml.DeviceGetVgpuSchedulerCapabilities
DeviceGetVgpuSchedulerLog = libnvml.DeviceGetVgpuSchedulerLog
DeviceGetVgpuSchedulerState = libnvml.DeviceGetVgpuSchedulerState
DeviceGetVgpuTypeCreatablePlacements = libnvml.DeviceGetVgpuTypeCreatablePlacements
DeviceGetVgpuTypeSupportedPlacements = libnvml.DeviceGetVgpuTypeSupportedPlacements
DeviceGetVgpuUtilization = libnvml.DeviceGetVgpuUtilization
DeviceGetViolationStatus = libnvml.DeviceGetViolationStatus
DeviceGetVirtualizationMode = libnvml.DeviceGetVirtualizationMode
@ -207,6 +227,7 @@ var (
DeviceSetApplicationsClocks = libnvml.DeviceSetApplicationsClocks
DeviceSetAutoBoostedClocksEnabled = libnvml.DeviceSetAutoBoostedClocksEnabled
DeviceSetComputeMode = libnvml.DeviceSetComputeMode
DeviceSetConfComputeUnprotectedMemSize = libnvml.DeviceSetConfComputeUnprotectedMemSize
DeviceSetCpuAffinity = libnvml.DeviceSetCpuAffinity
DeviceSetDefaultAutoBoostedClocksEnabled = libnvml.DeviceSetDefaultAutoBoostedClocksEnabled
DeviceSetDefaultFanSpeed_v2 = libnvml.DeviceSetDefaultFanSpeed_v2
@ -224,7 +245,10 @@ var (
DeviceSetNvLinkUtilizationControl = libnvml.DeviceSetNvLinkUtilizationControl
DeviceSetPersistenceMode = libnvml.DeviceSetPersistenceMode
DeviceSetPowerManagementLimit = libnvml.DeviceSetPowerManagementLimit
DeviceSetPowerManagementLimit_v2 = libnvml.DeviceSetPowerManagementLimit_v2
DeviceSetTemperatureThreshold = libnvml.DeviceSetTemperatureThreshold
DeviceSetVgpuCapabilities = libnvml.DeviceSetVgpuCapabilities
DeviceSetVgpuHeterogeneousMode = libnvml.DeviceSetVgpuHeterogeneousMode
DeviceSetVgpuSchedulerState = libnvml.DeviceSetVgpuSchedulerState
DeviceSetVirtualizationMode = libnvml.DeviceSetVirtualizationMode
DeviceValidateInforom = libnvml.DeviceValidateInforom
@ -243,9 +267,11 @@ var (
GpmMigSampleGet = libnvml.GpmMigSampleGet
GpmQueryDeviceSupport = libnvml.GpmQueryDeviceSupport
GpmQueryDeviceSupportV = libnvml.GpmQueryDeviceSupportV
GpmQueryIfStreamingEnabled = libnvml.GpmQueryIfStreamingEnabled
GpmSampleAlloc = libnvml.GpmSampleAlloc
GpmSampleFree = libnvml.GpmSampleFree
GpmSampleGet = libnvml.GpmSampleGet
GpmSetStreamingEnabled = libnvml.GpmSetStreamingEnabled
GpuInstanceCreateComputeInstance = libnvml.GpuInstanceCreateComputeInstance
GpuInstanceCreateComputeInstanceWithPlacement = libnvml.GpuInstanceCreateComputeInstanceWithPlacement
GpuInstanceDestroy = libnvml.GpuInstanceDestroy
@ -260,6 +286,9 @@ var (
InitWithFlags = libnvml.InitWithFlags
SetVgpuVersion = libnvml.SetVgpuVersion
Shutdown = libnvml.Shutdown
SystemGetConfComputeCapabilities = libnvml.SystemGetConfComputeCapabilities
SystemGetConfComputeKeyRotationThresholdInfo = libnvml.SystemGetConfComputeKeyRotationThresholdInfo
SystemGetConfComputeSettings = libnvml.SystemGetConfComputeSettings
SystemGetCudaDriverVersion = libnvml.SystemGetCudaDriverVersion
SystemGetCudaDriverVersion_v2 = libnvml.SystemGetCudaDriverVersion_v2
SystemGetDriverVersion = libnvml.SystemGetDriverVersion
@ -267,6 +296,7 @@ var (
SystemGetNVMLVersion = libnvml.SystemGetNVMLVersion
SystemGetProcessName = libnvml.SystemGetProcessName
SystemGetTopologyGpuSet = libnvml.SystemGetTopologyGpuSet
SystemSetConfComputeKeyRotationThresholdInfo = libnvml.SystemSetConfComputeKeyRotationThresholdInfo
UnitGetCount = libnvml.UnitGetCount
UnitGetDevices = libnvml.UnitGetDevices
UnitGetFanSpeedInfo = libnvml.UnitGetFanSpeedInfo
@ -319,8 +349,6 @@ var (
type Interface interface {
ComputeInstanceDestroy(ComputeInstance) Return
ComputeInstanceGetInfo(ComputeInstance) (ComputeInstanceInfo, Return)
DeviceCcuGetStreamState(Device) (int, Return)
DeviceCcuSetStreamState(Device, int) Return
DeviceClearAccountingPids(Device) Return
DeviceClearCpuAffinity(Device) Return
DeviceClearEccErrorCounts(Device, EccCounterType) Return
@ -346,12 +374,17 @@ type Interface interface {
DeviceGetBrand(Device) (BrandType, Return)
DeviceGetBridgeChipInfo(Device) (BridgeChipHierarchy, Return)
DeviceGetBusType(Device) (BusType, Return)
DeviceGetC2cModeInfoV(Device) C2cModeInfoHandler
DeviceGetClkMonStatus(Device) (ClkMonStatus, Return)
DeviceGetClock(Device, ClockType, ClockId) (uint32, Return)
DeviceGetClockInfo(Device, ClockType) (uint32, Return)
DeviceGetComputeInstanceId(Device) (int, Return)
DeviceGetComputeMode(Device) (ComputeMode, Return)
DeviceGetComputeRunningProcesses(Device) ([]ProcessInfo, Return)
DeviceGetConfComputeGpuAttestationReport(Device) (ConfComputeGpuAttestationReport, Return)
DeviceGetConfComputeGpuCertificate(Device) (ConfComputeGpuCertificate, Return)
DeviceGetConfComputeMemSizeInfo(Device) (ConfComputeMemSizeInfo, Return)
DeviceGetConfComputeProtectedMemoryUsage(Device) (Memory, Return)
DeviceGetCount() (int, Return)
DeviceGetCpuAffinity(Device, int) ([]uint, Return)
DeviceGetCpuAffinityWithinScope(Device, int, AffinityScope) ([]uint, Return)
@ -359,6 +392,7 @@ type Interface interface {
DeviceGetCudaComputeCapability(Device) (int, int, Return)
DeviceGetCurrPcieLinkGeneration(Device) (int, Return)
DeviceGetCurrPcieLinkWidth(Device) (int, Return)
DeviceGetCurrentClocksEventReasons(Device) (uint64, Return)
DeviceGetCurrentClocksThrottleReasons(Device) (uint64, Return)
DeviceGetDecoderUtilization(Device) (uint32, uint32, Return)
DeviceGetDefaultApplicationsClock(Device, ClockType) (uint32, Return)
@ -384,11 +418,12 @@ type Interface interface {
DeviceGetGpcClkMinMaxVfOffset(Device) (int, int, Return)
DeviceGetGpcClkVfOffset(Device) (int, Return)
DeviceGetGpuFabricInfo(Device) (GpuFabricInfo, Return)
DeviceGetGpuFabricInfoV(Device) GpuFabricInfoHandler
DeviceGetGpuInstanceById(Device, int) (GpuInstance, Return)
DeviceGetGpuInstanceId(Device) (int, Return)
DeviceGetGpuInstancePossiblePlacements(Device, *GpuInstanceProfileInfo) ([]GpuInstancePlacement, Return)
DeviceGetGpuInstanceProfileInfo(Device, int) (GpuInstanceProfileInfo, Return)
DeviceGetGpuInstanceProfileInfoV(Device, int) GpuInstanceProfileInfoV
DeviceGetGpuInstanceProfileInfoV(Device, int) GpuInstanceProfileInfoHandler
DeviceGetGpuInstanceRemainingCapacity(Device, *GpuInstanceProfileInfo) (int, Return)
DeviceGetGpuInstances(Device, *GpuInstanceProfileInfo) ([]GpuInstance, Return)
DeviceGetGpuMaxPcieLinkGeneration(Device) (int, Return)
@ -407,6 +442,8 @@ type Interface interface {
DeviceGetInforomImageVersion(Device) (string, Return)
DeviceGetInforomVersion(Device, InforomObject) (string, Return)
DeviceGetIrqNum(Device) (int, Return)
DeviceGetJpgUtilization(Device) (uint32, uint32, Return)
DeviceGetLastBBXFlushTime(Device) (uint64, uint, Return)
DeviceGetMPSComputeRunningProcesses(Device) ([]ProcessInfo, Return)
DeviceGetMaxClockInfo(Device, ClockType) (uint32, Return)
DeviceGetMaxCustomerBoostClock(Device, ClockType) (uint32, Return)
@ -425,10 +462,12 @@ type Interface interface {
DeviceGetMinMaxClockOfPState(Device, ClockType, Pstates) (uint32, uint32, Return)
DeviceGetMinMaxFanSpeed(Device) (int, int, Return)
DeviceGetMinorNumber(Device) (int, Return)
DeviceGetModuleId(Device) (int, Return)
DeviceGetMultiGpuBoard(Device) (int, Return)
DeviceGetName(Device) (string, Return)
DeviceGetNumFans(Device) (int, Return)
DeviceGetNumGpuCores(Device) (int, Return)
DeviceGetNumaNodeId(Device) (int, Return)
DeviceGetNvLinkCapability(Device, int, NvLinkCapability) (uint32, Return)
DeviceGetNvLinkErrorCounter(Device, int, NvLinkErrorCounter) (uint64, Return)
DeviceGetNvLinkRemoteDeviceType(Device, int) (IntNvLinkDeviceType, Return)
@ -437,8 +476,10 @@ type Interface interface {
DeviceGetNvLinkUtilizationControl(Device, int, int) (NvLinkUtilizationControl, Return)
DeviceGetNvLinkUtilizationCounter(Device, int, int) (uint64, uint64, Return)
DeviceGetNvLinkVersion(Device, int) (uint32, Return)
DeviceGetOfaUtilization(Device) (uint32, uint32, Return)
DeviceGetP2PStatus(Device, Device, GpuP2PCapsIndex) (GpuP2PStatus, Return)
DeviceGetPciInfo(Device) (PciInfo, Return)
DeviceGetPciInfoExt(Device) (PciInfoExt, Return)
DeviceGetPcieLinkMaxSpeed(Device) (uint32, Return)
DeviceGetPcieReplayCounter(Device) (int, Return)
DeviceGetPcieSpeed(Device) (int, Return)
@ -454,13 +495,17 @@ type Interface interface {
DeviceGetPowerState(Device) (Pstates, Return)
DeviceGetPowerUsage(Device) (uint32, Return)
DeviceGetProcessUtilization(Device, uint64) ([]ProcessUtilizationSample, Return)
DeviceGetProcessesUtilizationInfo(Device) (ProcessesUtilizationInfo, Return)
DeviceGetRemappedRows(Device) (int, int, bool, bool, Return)
DeviceGetRetiredPages(Device, PageRetirementCause) ([]uint64, Return)
DeviceGetRetiredPagesPendingStatus(Device) (EnableState, Return)
DeviceGetRetiredPages_v2(Device, PageRetirementCause) ([]uint64, []uint64, Return)
DeviceGetRowRemapperHistogram(Device) (RowRemapperHistogramValues, Return)
DeviceGetRunningProcessDetailList(Device) (ProcessDetailList, Return)
DeviceGetSamples(Device, SamplingType, uint64) (ValueType, []Sample, Return)
DeviceGetSerial(Device) (string, Return)
DeviceGetSramEccErrorStatus(Device) (EccSramErrorStatus, Return)
DeviceGetSupportedClocksEventReasons(Device) (uint64, Return)
DeviceGetSupportedClocksThrottleReasons(Device) (uint64, Return)
DeviceGetSupportedEventTypes(Device) (uint64, Return)
DeviceGetSupportedGraphicsClocks(Device, int) (int, uint32, Return)
@ -479,11 +524,16 @@ type Interface interface {
DeviceGetUtilizationRates(Device) (Utilization, Return)
DeviceGetVbiosVersion(Device) (string, Return)
DeviceGetVgpuCapabilities(Device, DeviceVgpuCapability) (bool, Return)
DeviceGetVgpuHeterogeneousMode(Device) (VgpuHeterogeneousMode, Return)
DeviceGetVgpuInstancesUtilizationInfo(Device) (VgpuInstancesUtilizationInfo, Return)
DeviceGetVgpuMetadata(Device) (VgpuPgpuMetadata, Return)
DeviceGetVgpuProcessUtilization(Device, uint64) ([]VgpuProcessUtilizationSample, Return)
DeviceGetVgpuProcessesUtilizationInfo(Device) (VgpuProcessesUtilizationInfo, Return)
DeviceGetVgpuSchedulerCapabilities(Device) (VgpuSchedulerCapabilities, Return)
DeviceGetVgpuSchedulerLog(Device) (VgpuSchedulerLog, Return)
DeviceGetVgpuSchedulerState(Device) (VgpuSchedulerGetState, Return)
DeviceGetVgpuTypeCreatablePlacements(Device, VgpuTypeId) (VgpuPlacementList, Return)
DeviceGetVgpuTypeSupportedPlacements(Device, VgpuTypeId) (VgpuPlacementList, Return)
DeviceGetVgpuUtilization(Device, uint64) (ValueType, []VgpuInstanceUtilizationSample, Return)
DeviceGetViolationStatus(Device, PerfPolicyType) (ViolationTime, Return)
DeviceGetVirtualizationMode(Device) (GpuVirtualizationMode, Return)
@ -504,6 +554,7 @@ type Interface interface {
DeviceSetApplicationsClocks(Device, uint32, uint32) Return
DeviceSetAutoBoostedClocksEnabled(Device, EnableState) Return
DeviceSetComputeMode(Device, ComputeMode) Return
DeviceSetConfComputeUnprotectedMemSize(Device, uint64) Return
DeviceSetCpuAffinity(Device) Return
DeviceSetDefaultAutoBoostedClocksEnabled(Device, EnableState, uint32) Return
DeviceSetDefaultFanSpeed_v2(Device, int) Return
@ -521,7 +572,10 @@ type Interface interface {
DeviceSetNvLinkUtilizationControl(Device, int, int, *NvLinkUtilizationControl, bool) Return
DeviceSetPersistenceMode(Device, EnableState) Return
DeviceSetPowerManagementLimit(Device, uint32) Return
DeviceSetPowerManagementLimit_v2(Device, *PowerValue_v2) Return
DeviceSetTemperatureThreshold(Device, TemperatureThresholds, int) Return
DeviceSetVgpuCapabilities(Device, DeviceVgpuCapability, EnableState) Return
DeviceSetVgpuHeterogeneousMode(Device, VgpuHeterogeneousMode) Return
DeviceSetVgpuSchedulerState(Device, *VgpuSchedulerSetState) Return
DeviceSetVirtualizationMode(Device, GpuVirtualizationMode) Return
DeviceValidateInforom(Device) Return
@ -540,16 +594,18 @@ type Interface interface {
GpmMigSampleGet(Device, int, GpmSample) Return
GpmQueryDeviceSupport(Device) (GpmSupport, Return)
GpmQueryDeviceSupportV(Device) GpmSupportV
GpmQueryIfStreamingEnabled(Device) (uint32, Return)
GpmSampleAlloc() (GpmSample, Return)
GpmSampleFree(GpmSample) Return
GpmSampleGet(Device, GpmSample) Return
GpmSetStreamingEnabled(Device, uint32) Return
GpuInstanceCreateComputeInstance(GpuInstance, *ComputeInstanceProfileInfo) (ComputeInstance, Return)
GpuInstanceCreateComputeInstanceWithPlacement(GpuInstance, *ComputeInstanceProfileInfo, *ComputeInstancePlacement) (ComputeInstance, Return)
GpuInstanceDestroy(GpuInstance) Return
GpuInstanceGetComputeInstanceById(GpuInstance, int) (ComputeInstance, Return)
GpuInstanceGetComputeInstancePossiblePlacements(GpuInstance, *ComputeInstanceProfileInfo) ([]ComputeInstancePlacement, Return)
GpuInstanceGetComputeInstanceProfileInfo(GpuInstance, int, int) (ComputeInstanceProfileInfo, Return)
GpuInstanceGetComputeInstanceProfileInfoV(GpuInstance, int, int) ComputeInstanceProfileInfoV
GpuInstanceGetComputeInstanceProfileInfoV(GpuInstance, int, int) ComputeInstanceProfileInfoHandler
GpuInstanceGetComputeInstanceRemainingCapacity(GpuInstance, *ComputeInstanceProfileInfo) (int, Return)
GpuInstanceGetComputeInstances(GpuInstance, *ComputeInstanceProfileInfo) ([]ComputeInstance, Return)
GpuInstanceGetInfo(GpuInstance) (GpuInstanceInfo, Return)
@ -557,6 +613,9 @@ type Interface interface {
InitWithFlags(uint32) Return
SetVgpuVersion(*VgpuVersion) Return
Shutdown() Return
SystemGetConfComputeCapabilities() (ConfComputeSystemCaps, Return)
SystemGetConfComputeKeyRotationThresholdInfo() (ConfComputeGetKeyRotationThresholdInfo, Return)
SystemGetConfComputeSettings() (SystemConfComputeSettings, Return)
SystemGetCudaDriverVersion() (int, Return)
SystemGetCudaDriverVersion_v2() (int, Return)
SystemGetDriverVersion() (string, Return)
@ -564,6 +623,7 @@ type Interface interface {
SystemGetNVMLVersion() (string, Return)
SystemGetProcessName(int) (string, Return)
SystemGetTopologyGpuSet(int) ([]Device, Return)
SystemSetConfComputeKeyRotationThresholdInfo(ConfComputeSetKeyRotationThresholdInfo) Return
UnitGetCount() (int, Return)
UnitGetDevices(Unit) ([]Device, Return)
UnitGetFanSpeedInfo(Unit) (UnitFanSpeeds, Return)
@ -614,8 +674,6 @@ type Interface interface {
//
//go:generate moq -out mock/device.go -pkg mock . Device:Device
type Device interface {
CcuGetStreamState() (int, Return)
CcuSetStreamState(int) Return
ClearAccountingPids() Return
ClearCpuAffinity() Return
ClearEccErrorCounts(EccCounterType) Return
@ -640,18 +698,24 @@ type Device interface {
GetBrand() (BrandType, Return)
GetBridgeChipInfo() (BridgeChipHierarchy, Return)
GetBusType() (BusType, Return)
GetC2cModeInfoV() C2cModeInfoHandler
GetClkMonStatus() (ClkMonStatus, Return)
GetClock(ClockType, ClockId) (uint32, Return)
GetClockInfo(ClockType) (uint32, Return)
GetComputeInstanceId() (int, Return)
GetComputeMode() (ComputeMode, Return)
GetComputeRunningProcesses() ([]ProcessInfo, Return)
GetConfComputeGpuAttestationReport() (ConfComputeGpuAttestationReport, Return)
GetConfComputeGpuCertificate() (ConfComputeGpuCertificate, Return)
GetConfComputeMemSizeInfo() (ConfComputeMemSizeInfo, Return)
GetConfComputeProtectedMemoryUsage() (Memory, Return)
GetCpuAffinity(int) ([]uint, Return)
GetCpuAffinityWithinScope(int, AffinityScope) ([]uint, Return)
GetCreatableVgpus() ([]VgpuTypeId, Return)
GetCudaComputeCapability() (int, int, Return)
GetCurrPcieLinkGeneration() (int, Return)
GetCurrPcieLinkWidth() (int, Return)
GetCurrentClocksEventReasons() (uint64, Return)
GetCurrentClocksThrottleReasons() (uint64, Return)
GetDecoderUtilization() (uint32, uint32, Return)
GetDefaultApplicationsClock(ClockType) (uint32, Return)
@ -677,11 +741,12 @@ type Device interface {
GetGpcClkMinMaxVfOffset() (int, int, Return)
GetGpcClkVfOffset() (int, Return)
GetGpuFabricInfo() (GpuFabricInfo, Return)
GetGpuFabricInfoV() GpuFabricInfoHandler
GetGpuInstanceById(int) (GpuInstance, Return)
GetGpuInstanceId() (int, Return)
GetGpuInstancePossiblePlacements(*GpuInstanceProfileInfo) ([]GpuInstancePlacement, Return)
GetGpuInstanceProfileInfo(int) (GpuInstanceProfileInfo, Return)
GetGpuInstanceProfileInfoV(int) GpuInstanceProfileInfoV
GetGpuInstanceProfileInfoV(int) GpuInstanceProfileInfoHandler
GetGpuInstanceRemainingCapacity(*GpuInstanceProfileInfo) (int, Return)
GetGpuInstances(*GpuInstanceProfileInfo) ([]GpuInstance, Return)
GetGpuMaxPcieLinkGeneration() (int, Return)
@ -696,6 +761,8 @@ type Device interface {
GetInforomImageVersion() (string, Return)
GetInforomVersion(InforomObject) (string, Return)
GetIrqNum() (int, Return)
GetJpgUtilization() (uint32, uint32, Return)
GetLastBBXFlushTime() (uint64, uint, Return)
GetMPSComputeRunningProcesses() ([]ProcessInfo, Return)
GetMaxClockInfo(ClockType) (uint32, Return)
GetMaxCustomerBoostClock(ClockType) (uint32, Return)
@ -714,10 +781,12 @@ type Device interface {
GetMinMaxClockOfPState(ClockType, Pstates) (uint32, uint32, Return)
GetMinMaxFanSpeed() (int, int, Return)
GetMinorNumber() (int, Return)
GetModuleId() (int, Return)
GetMultiGpuBoard() (int, Return)
GetName() (string, Return)
GetNumFans() (int, Return)
GetNumGpuCores() (int, Return)
GetNumaNodeId() (int, Return)
GetNvLinkCapability(int, NvLinkCapability) (uint32, Return)
GetNvLinkErrorCounter(int, NvLinkErrorCounter) (uint64, Return)
GetNvLinkRemoteDeviceType(int) (IntNvLinkDeviceType, Return)
@ -726,8 +795,10 @@ type Device interface {
GetNvLinkUtilizationControl(int, int) (NvLinkUtilizationControl, Return)
GetNvLinkUtilizationCounter(int, int) (uint64, uint64, Return)
GetNvLinkVersion(int) (uint32, Return)
GetOfaUtilization() (uint32, uint32, Return)
GetP2PStatus(Device, GpuP2PCapsIndex) (GpuP2PStatus, Return)
GetPciInfo() (PciInfo, Return)
GetPciInfoExt() (PciInfoExt, Return)
GetPcieLinkMaxSpeed() (uint32, Return)
GetPcieReplayCounter() (int, Return)
GetPcieSpeed() (int, Return)
@ -743,13 +814,17 @@ type Device interface {
GetPowerState() (Pstates, Return)
GetPowerUsage() (uint32, Return)
GetProcessUtilization(uint64) ([]ProcessUtilizationSample, Return)
GetProcessesUtilizationInfo() (ProcessesUtilizationInfo, Return)
GetRemappedRows() (int, int, bool, bool, Return)
GetRetiredPages(PageRetirementCause) ([]uint64, Return)
GetRetiredPagesPendingStatus() (EnableState, Return)
GetRetiredPages_v2(PageRetirementCause) ([]uint64, []uint64, Return)
GetRowRemapperHistogram() (RowRemapperHistogramValues, Return)
GetRunningProcessDetailList() (ProcessDetailList, Return)
GetSamples(SamplingType, uint64) (ValueType, []Sample, Return)
GetSerial() (string, Return)
GetSramEccErrorStatus() (EccSramErrorStatus, Return)
GetSupportedClocksEventReasons() (uint64, Return)
GetSupportedClocksThrottleReasons() (uint64, Return)
GetSupportedEventTypes() (uint64, Return)
GetSupportedGraphicsClocks(int) (int, uint32, Return)
@ -768,18 +843,25 @@ type Device interface {
GetUtilizationRates() (Utilization, Return)
GetVbiosVersion() (string, Return)
GetVgpuCapabilities(DeviceVgpuCapability) (bool, Return)
GetVgpuHeterogeneousMode() (VgpuHeterogeneousMode, Return)
GetVgpuInstancesUtilizationInfo() (VgpuInstancesUtilizationInfo, Return)
GetVgpuMetadata() (VgpuPgpuMetadata, Return)
GetVgpuProcessUtilization(uint64) ([]VgpuProcessUtilizationSample, Return)
GetVgpuProcessesUtilizationInfo() (VgpuProcessesUtilizationInfo, Return)
GetVgpuSchedulerCapabilities() (VgpuSchedulerCapabilities, Return)
GetVgpuSchedulerLog() (VgpuSchedulerLog, Return)
GetVgpuSchedulerState() (VgpuSchedulerGetState, Return)
GetVgpuTypeCreatablePlacements(VgpuTypeId) (VgpuPlacementList, Return)
GetVgpuTypeSupportedPlacements(VgpuTypeId) (VgpuPlacementList, Return)
GetVgpuUtilization(uint64) (ValueType, []VgpuInstanceUtilizationSample, Return)
GetViolationStatus(PerfPolicyType) (ViolationTime, Return)
GetVirtualizationMode() (GpuVirtualizationMode, Return)
GpmMigSampleGet(int, GpmSample) Return
GpmQueryDeviceSupport() (GpmSupport, Return)
GpmQueryDeviceSupportV() GpmSupportV
GpmQueryIfStreamingEnabled() (uint32, Return)
GpmSampleGet(GpmSample) Return
GpmSetStreamingEnabled(uint32) Return
IsMigDeviceHandle() (bool, Return)
OnSameBoard(Device) (int, Return)
RegisterEvents(uint64, EventSet) Return
@ -793,6 +875,7 @@ type Device interface {
SetApplicationsClocks(uint32, uint32) Return
SetAutoBoostedClocksEnabled(EnableState) Return
SetComputeMode(ComputeMode) Return
SetConfComputeUnprotectedMemSize(uint64) Return
SetCpuAffinity() Return
SetDefaultAutoBoostedClocksEnabled(EnableState, uint32) Return
SetDefaultFanSpeed_v2(int) Return
@ -810,7 +893,10 @@ type Device interface {
SetNvLinkUtilizationControl(int, int, *NvLinkUtilizationControl, bool) Return
SetPersistenceMode(EnableState) Return
SetPowerManagementLimit(uint32) Return
SetPowerManagementLimit_v2(*PowerValue_v2) Return
SetTemperatureThreshold(TemperatureThresholds, int) Return
SetVgpuCapabilities(DeviceVgpuCapability, EnableState) Return
SetVgpuHeterogeneousMode(VgpuHeterogeneousMode) Return
SetVgpuSchedulerState(*VgpuSchedulerSetState) Return
SetVirtualizationMode(GpuVirtualizationMode) Return
ValidateInforom() Return
@ -827,7 +913,7 @@ type GpuInstance interface {
GetComputeInstanceById(int) (ComputeInstance, Return)
GetComputeInstancePossiblePlacements(*ComputeInstanceProfileInfo) ([]ComputeInstancePlacement, Return)
GetComputeInstanceProfileInfo(int, int) (ComputeInstanceProfileInfo, Return)
GetComputeInstanceProfileInfoV(int, int) ComputeInstanceProfileInfoV
GetComputeInstanceProfileInfoV(int, int) ComputeInstanceProfileInfoHandler
GetComputeInstanceRemainingCapacity(*ComputeInstanceProfileInfo) (int, Return)
GetComputeInstances(*ComputeInstanceProfileInfo) ([]ComputeInstance, Return)
GetInfo() (GpuInstanceInfo, Return)
@ -906,6 +992,7 @@ type VgpuInstance interface {
type VgpuTypeId interface {
GetCapabilities(VgpuCapability) (bool, Return)
GetClass() (string, Return)
GetCreatablePlacements(Device) (VgpuPlacementList, Return)
GetDeviceID() (uint64, uint64, Return)
GetFrameRateLimit() (uint32, Return)
GetFramebufferSize() (uint64, Return)
@ -916,4 +1003,5 @@ type VgpuTypeId interface {
GetName() (string, Return)
GetNumDisplayHeads() (int, Return)
GetResolution(int) (uint32, uint32, Return)
GetSupportedPlacements(Device) (VgpuPlacementList, Return)
}

2
vendor/modules.txt vendored
View File

@ -6,7 +6,7 @@ github.com/NVIDIA/go-nvlib/pkg/nvpci
github.com/NVIDIA/go-nvlib/pkg/nvpci/bytes
github.com/NVIDIA/go-nvlib/pkg/nvpci/mmio
github.com/NVIDIA/go-nvlib/pkg/pciids
# github.com/NVIDIA/go-nvml v0.12.0-6
# github.com/NVIDIA/go-nvml v0.12.4-0
## explicit; go 1.20
github.com/NVIDIA/go-nvml/pkg/dl
github.com/NVIDIA/go-nvml/pkg/nvml