def dcgmVgpuConfigGet(dcgm_handle, group_id, reqCfgType, count, status_handle):
fn = dcgm_structs._dcgmGetFunctionPointer("dcgmVgpuConfigGet")
vgpu_config_values_array = count * dcgm_structs.c_dcgmDeviceVgpuConfig_v1
c_config_values = vgpu_config_values_array()
for index in range(0, count):
c_config_values[
index].version = dcgm_structs.dcgmDeviceVgpuConfig_version1
ret = fn(dcgm_handle, group_id, reqCfgType, count, c_config_values,
status_handle)
dcgm_structs._dcgmCheckReturn(ret)
return map(None, c_config_values[0:count])
def vtDcgmVgpuConfigGet(dcgm_handle, group_id, reqCfgType, count,
status_handle, versionTest):
fn = dcgm_structs._dcgmGetFunctionPointer("dcgmVgpuConfigSet")
vgpu_config_values_array = count * dcgm_structs.c_dcgmDeviceVgpuConfig_v1
c_config_values = vgpu_config_values_array()
vgpuConfig = dcgm_structs.c_dcgmDeviceVgpuConfig_v1()
vgpuConfig.version = dcgm_structs.make_dcgm_version(vgpuConfig, 1)
logger.debug("Structure version: %d" % vgpuConfig.version)
for index in range(0, count):
c_config_values[index].version = versionTest
ret = fn(dcgm_handle, group_id, c_config_values, status_handle)
dcgm_structs._dcgmCheckReturn(ret)
return map(None, c_config_values[0:count])
def dcgmErrorGetFormatMsgByCode(code):
fn = dcgm_structs._dcgmGetFunctionPointer("dcgmErrorGetFormatMsgByCode")
fn.restype = ctypes.c_char_p
ret = fn(code)
return ret
def dcgmErrorGetPriorityByCode(code):
fn = dcgm_structs._dcgmGetFunctionPointer("dcgmErrorGetPriorityByCode")
ret = fn(code)
return ret
def dcgmErrorGetFormatMsgByCode(code):
fn = dcgm_structs._dcgmGetFunctionPointer("dcgmErrorGetFormatMsgByCode")
fn.restype = ctypes.c_char_p
ret = fn(code)
return ret.decode('utf-8') if isinstance(ret, bytes) else ret
def dcgmInjectEntityFieldValue(dcgmHandle, entityGroupId, entityId, value):
fn = dcgm_structs._dcgmGetFunctionPointer("dcgmInjectEntityFieldValue")
ret = fn(dcgmHandle, c_uint(entityGroupId), c_uint(entityId), byref(value))
_dcgmIntCheckReturn(ret)
return ret
def dcgmInjectFieldValue(dcgmHandle, gpuId, value):
fn = dcgm_structs._dcgmGetFunctionPointer("dcgmInjectFieldValue")
ret = fn(dcgmHandle, c_uint(gpuId), byref(value))
_dcgmIntCheckReturn(ret)
return ret
def dcgmUnwatchFieldValue(dcgmHandle, gpuId, fieldId, clearCache):
fn = dcgm_structs._dcgmGetFunctionPointer("dcgmUnwatchFieldValue")
ret = fn(dcgmHandle, c_int(gpuId), c_uint(fieldId), c_int(clearCache))
_dcgmIntCheckReturn(ret)
return ret
def dcgmServerRun(portNumber, socketPath, isConnectionTcp):
fn = dcgm_structs._dcgmGetFunctionPointer("dcgmEngineRun")
ret = fn(portNumber, socketPath, isConnectionTcp)
_dcgmIntCheckReturn(ret)
return ret
def dcgmStopDiagnostic(dcgm_handle):
fn = dcgm_structs._dcgmGetFunctionPointer("dcgmStopDiagnostic")
ret = fn(dcgm_handle)
dcgm_structs._dcgmCheckReturn(ret)
return ret
def dcgmVgpuConfigEnforce(dcgm_handle, group_id, status_handle):
fn = dcgm_structs._dcgmGetFunctionPointer("dcgmVgpuConfigEnforce")
ret = fn(dcgm_handle, group_id, status_handle)
dcgm_structs._dcgmCheckReturn(ret)
return ret
def dcgmVgpuConfigSet(dcgm_handle, group_id, configToSet, status_handle):
fn = dcgm_structs._dcgmGetFunctionPointer("dcgmVgpuConfigSet")
configToSet.version = dcgm_structs.dcgmDeviceVgpuConfig_version1
ret = fn(dcgm_handle, group_id, byref(configToSet), status_handle)
dcgm_structs._dcgmCheckReturn(ret)
return ret
def dcgmMetadataStateSetRunInterval(dcgmHandle, runIntervalMs):
fn = dcgm_structs._dcgmGetFunctionPointer(
"dcgmMetadataStateSetRunInterval")
ret = fn(dcgmHandle, c_uint(runIntervalMs))
_dcgmIntCheckReturn(ret)
def dcgmWatchFieldValue(dcgmHandle, gpuId, fieldId, updateFreq, maxKeepAge, maxKeepEntries):
fn = dcgm_structs._dcgmGetFunctionPointer("dcgmWatchFieldValue")
ret = fn(dcgmHandle, c_int(gpuId), c_uint(fieldId), c_longlong(updateFreq), c_double(maxKeepAge), c_int(maxKeepEntries))
_dcgmIntCheckReturn(ret)
return ret