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