def vtDcgmIntrospectGetFieldsExecTime(dcgm_handle, introspectContext, versionTest, waitIfNoData=True): fn = dcgmFP("dcgmIntrospectGetFieldsExecTime") execTime = dcgm_structs.c_dcgmIntrospectFieldsExecTime_v1() execTime.version = dcgm_structs.make_dcgm_version(execTime, 1) logger.debug("Structure version: %d" % execTime.version) fullExecTime = dcgm_structs.c_dcgmIntrospectFullFieldsExecTime_v2() fullExecTime.version = dcgm_structs.make_dcgm_version(fullExecTime, 2) logger.debug("Structure version: %d" % fullExecTime.version) fullExecTime.version = versionTest execTime.version = versionTest introspectContext = dcgm_structs.c_dcgmIntrospectContext_v1() introspectContext.version = dcgm_structs.make_dcgm_version( introspectContext, 1) logger.debug("Structure version: %d" % introspectContext.version) introspectContext.version = versionTest ret = fn(dcgm_handle, byref(introspectContext), byref(execTime), waitIfNoData) dcgm_structs._dcgmCheckReturn(ret) return execTime
def vtDcgmConnect_v2(ip_address, connectParams, versionTest): connectParams = dcgm_structs.c_dcgmConnectV2Params_v1() connectParams.version = dcgm_structs.make_dcgm_version(connectParams, 1) logger.debug("Structure version: %d" % connectParams.version) connectParams.version = versionTest dcgm_handle = c_void_p() fn = dcgmFP("dcgmConnect_v2") ret = fn(ip_address, byref(connectParams), byref(dcgm_handle)) dcgm_structs._dcgmCheckReturn(ret) return dcgm_handle
def vtDcgmGroupGetInfo(dcgm_handle, group_id, versionTest): fn = dcgmFP("dcgmGroupGetInfo") device_values = dcgm_structs.c_dcgmGroupInfo_v2() device_values.version = dcgm_structs.make_dcgm_version(device_values, 2) logger.debug("Structure version: %d" % device_values.version) device_values.version = versionTest ret = fn(dcgm_handle, group_id, byref(device_values)) dcgm_structs._dcgmCheckReturn(ret) return device_values
def vtDcgmConfigSet(dcgm_handle, group_id, configToSet, status_handle, versionTest): fn = dcgmFP("dcgmConfigSet") config_values = dcgm_structs.c_dcgmDeviceConfig_v1() config_values.version = dcgm_structs.make_dcgm_version(config_values, 1) logger.debug("Structure version: %d" % config_values.version) configToSet.version = versionTest ret = fn(dcgm_handle, group_id, byref(configToSet), status_handle) dcgm_structs._dcgmCheckReturn(ret) return ret
def vtDcgmHealthCheck(dcgm_handle, groupId, versionTest): c_results = dcgm_structs.c_dcgmHealthResponse_v4() c_results.version = dcgm_structs.make_dcgm_version(c_results, 4) logger.debug("Structure version: %d" % c_results.version) c_results.version = versionTest fn = dcgmFP("dcgmHealthCheck") ret = fn(dcgm_handle, groupId, byref(c_results)) dcgm_structs._dcgmCheckReturn(ret) return c_results
def vtDcgmFieldGroupGetAll(dcgm_handle, versionTest): c_allGroupInfo = dcgm_structs.c_dcgmAllFieldGroup_v1() c_allGroupInfo.version = dcgm_structs.make_dcgm_version(c_allGroupInfo, 1) logger.debug("Structure version: %d" % c_allGroupInfo.version) c_allGroupInfo.version = versionTest fn = dcgmFP("dcgmFieldGroupGetAll") ret = fn(dcgm_handle, byref(c_allGroupInfo)) dcgm_structs._dcgmCheckReturn(ret) return c_allGroupInfo
def vtDcgmGetDeviceAttributes(dcgm_handle, gpuId, versionTest): fn = dcgmFP("dcgmGetDeviceAttributes") device_values = dcgm_structs.c_dcgmDeviceAttributes_v1() device_values.version = dcgm_structs.make_dcgm_version(device_values, 1) logger.debug("Structure version: %d" % device_values.version) device_values.version = versionTest ret = fn(dcgm_handle, c_int(gpuId), byref(device_values)) dcgm_structs._dcgmCheckReturn(ret) return device_values
def vtDcgmPolicyGet(dcgm_handle, group_id, count, status_handle, versionTest): fn = dcgmFP("dcgmPolicyGet") policy_array = count * dcgm_structs.c_dcgmPolicy_v1 c_policy_values = policy_array() policy = dcgm_structs.c_dcgmPolicy_v1() policy.version = dcgm_structs.make_dcgm_version(policy, 1) logger.debug("Structure version: %d" % policy.version) policyCallback = dcgm_structs.c_dcgmPolicyCallbackResponse_v1() policyCallback.version = dcgm_structs.make_dcgm_version(policyCallback, 1) logger.debug("Structure version: %d" % policyCallback.version) for index in range(0, count): c_policy_values[index].version = versionTest ret = fn(dcgm_handle, group_id, count, c_policy_values, status_handle) dcgm_structs._dcgmCheckReturn(ret) return c_policy_values[0:count]
def vtDcgmJobGetStats(dcgm_handle, jobid, versionTest): fn = dcgmFP("dcgmJobGetStats") jobInfo = dcgm_structs.c_dcgmJobInfo_v3() jobInfo.version = dcgm_structs.make_dcgm_version(jobInfo, 3) logger.debug("Structure version: %d" % jobInfo.version) jobInfo.version = versionTest ret = fn(dcgm_handle, jobid, byref(jobInfo)) dcgm_structs._dcgmCheckReturn(ret) return jobInfo
def vtDcgmFieldGroupGetInfo(dcgm_handle, fieldGroupId, versionTest): c_fieldGroupInfo = dcgm_structs.c_dcgmFieldGroupInfo_v1() c_fieldGroupInfo.version = dcgm_structs.make_dcgm_version( c_fieldGroupInfo, 1) logger.debug("Structure version: %d" % c_fieldGroupInfo.version) c_fieldGroupInfo.version = versionTest c_fieldGroupInfo.fieldGroupId = fieldGroupId fn = dcgmFP("dcgmFieldGroupGetInfo") ret = fn(dcgm_handle, byref(c_fieldGroupInfo)) dcgm_structs._dcgmCheckReturn(ret) return c_fieldGroupInfo
def vtDcgmGetPidInfo(dcgm_handle, groupId, pid, versionTest): fn = dcgmFP("dcgmGetPidInfo") pidInfo = dcgm_structs.c_dcgmPidInfo_v2() pidInfo.version = dcgm_structs.make_dcgm_version( dcgm_structs.c_dcgmPidInfo_v2, 2) logger.debug("Structure version: %d" % pidInfo.version) pidInfo.version = versionTest pidInfo.pid = pid ret = fn(dcgm_handle, groupId, byref(pidInfo)) dcgm_structs._dcgmCheckReturn(ret) return pidInfo
def vtDcgmIntrospectGetHostengineCpuUtilization(dcgm_handle, versionTest, waitIfNoData=True): fn = dcgmFP("dcgmIntrospectGetHostengineCpuUtilization") cpuUtil = dcgm_structs.c_dcgmIntrospectCpuUtil_v1() cpuUtil.version = dcgm_structs.make_dcgm_version(cpuUtil, 1) logger.debug("Structure version: %d" % cpuUtil.version) cpuUtil.version = versionTest ret = fn(dcgm_handle, byref(cpuUtil), waitIfNoData) dcgm_structs._dcgmCheckReturn(ret) return cpuUtil
def vtDcgmIntrospectGetHostengineMemoryUsage(dcgm_handle, versionTest, waitIfNoData=True): fn = dcgmFP("dcgmIntrospectGetHostengineMemoryUsage") memInfo = dcgm_structs.c_dcgmIntrospectMemory_v1() memInfo.version = dcgm_structs.make_dcgm_version(memInfo, 1) logger.debug("Structure version: %d" % memInfo.version) memInfo.version = versionTest ret = fn(dcgm_handle, byref(memInfo), waitIfNoData) dcgm_structs._dcgmCheckReturn(ret) return memInfo
def vtDcgmActionValidate_v2(dcgm_handle, runDiagInfo, versionTest): response = dcgm_structs.c_dcgmDiagResponse_v6() response.version = dcgm_structs.make_dcgm_version(response, 6) logger.debug("Structure version: %d" % response.version) runDiagInfo = dcgm_structs.c_dcgmRunDiag_v7() runDiagInfo.version = dcgm_structs.dcgmRunDiag_version7 logger.debug("Structure version: %d" % runDiagInfo.version) runDiagInfo.version = versionTest response.version = versionTest fn = dcgmFP("dcgmActionValidate_v2") ret = fn(dcgm_handle, byref(runDiagInfo), byref(response)) dcgm_structs._dcgmCheckReturn(ret) return response
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 vtDcgmIntrospectGetFieldsMemoryUsage(dcgm_handle, introspectContext, versionTest, waitIfNoData=True): fn = dcgmFP("dcgmIntrospectGetFieldsMemoryUsage") memInfo = dcgm_structs.c_dcgmIntrospectFullMemory_v1() memInfo.version = dcgm_structs.make_dcgm_version(memInfo, 1) logger.debug("Structure version: %d" % memInfo.version) memInfo.version = versionTest introspectContext = dcgm_structs.c_dcgmIntrospectContext_v1() introspectContext.version = versionTest ret = fn(dcgm_handle, byref(introspectContext), byref(memInfo), waitIfNoData) dcgm_structs._dcgmCheckReturn(ret) return memInfo
# version must always be first ('version', c_uint), ('a', c_uint) ] class c_dcgmVersionTest_v2(dcgm_structs._PrintableStructure): _fields_ = [ # version must always be first ('version', c_uint), ('a', c_uint), ('b', c_uint) ] dcgmVersionTest_version1 = dcgm_structs.make_dcgm_version( c_dcgmVersionTest_v1, 1) dcgmVersionTest_version2 = dcgm_structs.make_dcgm_version( c_dcgmVersionTest_v2, 2) dcgmVersionTest_version3 = dcgm_structs.make_dcgm_version( c_dcgmVersionTest_v2, 3) # Represents a command to save or load a JSON file to/from the DcgmCacheManager _dcgmStatsFileType_t = c_uint DCGM_STATS_FILE_TYPE_JSON = 0 class c_dcgmCacheManagerSave_v1(dcgm_structs._PrintableStructure): _fields_ = [ # version must always be first ('version', c_uint), ('fileType', _dcgmStatsFileType_t),