Marko Oblak
01474ff14e
Signed-off-by: Marko Oblak <Marko.Oblak@amd.com> Change-Id: I0cff925082e6bc637e4b5073df64445380b3a3f5
646 строки
26 KiB
C++
646 строки
26 KiB
C++
/*
|
|
* =============================================================================
|
|
* The University of Illinois/NCSA
|
|
* Open Source License (NCSA)
|
|
*
|
|
* Copyright (c) 2022, Advanced Micro Devices, Inc.
|
|
* All rights reserved.
|
|
*
|
|
* Developed by:
|
|
*
|
|
* AMD Research and AMD ROC Software Development
|
|
*
|
|
* Advanced Micro Devices, Inc.
|
|
*
|
|
* www.amd.com
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to
|
|
* deal with the Software without restriction, including without limitation
|
|
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
* and/or sell copies of the Software, and to permit persons to whom the
|
|
* Software is furnished to do so, subject to the following conditions:
|
|
*
|
|
* - Redistributions of source code must retain the above copyright notice,
|
|
* this list of conditions and the following disclaimers.
|
|
* - Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimers in
|
|
* the documentation and/or other materials provided with the distribution.
|
|
* - Neither the names of <Name of Development Group, Name of Institution>,
|
|
* nor the names of its contributors may be used to endorse or promote
|
|
* products derived from this Software without specific prior written
|
|
* permission.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
|
* THE CONTRIBUTORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
|
|
* OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
|
|
* ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
* DEALINGS WITH THE SOFTWARE.
|
|
*
|
|
*/
|
|
#include <assert.h>
|
|
#include <stdint.h>
|
|
#include <unistd.h>
|
|
|
|
#include "amd_smi/amdsmi.h"
|
|
#include <bitset>
|
|
#include <iostream>
|
|
#include <pwd.h>
|
|
#include <sys/stat.h>
|
|
#include <string.h>
|
|
#include <vector>
|
|
|
|
|
|
#define CHK_AMDSMI_RET(RET) \
|
|
{ \
|
|
if (RET != AMDSMI_STATUS_SUCCESS) { \
|
|
const char *err_str; \
|
|
std::cout << "AMDSMI call returned " << RET << " at line " \
|
|
<< __LINE__ << std::endl; \
|
|
amdsmi_status_code_to_string(RET, &err_str); \
|
|
std::cout << err_str << std::endl; \
|
|
return RET; \
|
|
} \
|
|
}
|
|
|
|
|
|
void getFWNameFromId(int id, char *name)
|
|
{
|
|
switch (id) {
|
|
case FW_ID_SMU:
|
|
strcpy(name, "SMU");
|
|
break;
|
|
case FW_ID_CP_CE:
|
|
strcpy(name, "CP_CE");
|
|
break;
|
|
case FW_ID_CP_PFP:
|
|
strcpy(name, "CP_PFP");
|
|
break;
|
|
case FW_ID_CP_ME:
|
|
strcpy(name, "CP_ME");
|
|
break;
|
|
case FW_ID_CP_MEC_JT1:
|
|
strcpy(name, "CP_MEC_JT1");
|
|
break;
|
|
case FW_ID_CP_MEC_JT2:
|
|
strcpy(name, "CP_MEC_JT2");
|
|
break;
|
|
case FW_ID_CP_MEC1:
|
|
strcpy(name, "CP_MEC1");
|
|
break;
|
|
case FW_ID_CP_MEC2:
|
|
strcpy(name, "CP_MEC2");
|
|
break;
|
|
case FW_ID_RLC:
|
|
strcpy(name, "RLC");
|
|
break;
|
|
case FW_ID_SDMA0:
|
|
strcpy(name, "SDMA0");
|
|
break;
|
|
case FW_ID_SDMA1:
|
|
strcpy(name, "SDMA1");
|
|
break;
|
|
case FW_ID_SDMA2:
|
|
strcpy(name, "SDMA2");
|
|
break;
|
|
case FW_ID_SDMA3:
|
|
strcpy(name, "SDMA3");
|
|
break;
|
|
case FW_ID_SDMA4:
|
|
strcpy(name, "SDMA4");
|
|
break;
|
|
case FW_ID_SDMA5:
|
|
strcpy(name, "SDMA5");
|
|
break;
|
|
case FW_ID_SDMA6:
|
|
strcpy(name, "SDMA6");
|
|
break;
|
|
case FW_ID_SDMA7:
|
|
strcpy(name, "SDMA7");
|
|
break;
|
|
case FW_ID_VCN:
|
|
strcpy(name, "VCN");
|
|
break;
|
|
case FW_ID_UVD:
|
|
strcpy(name, "UVD");
|
|
break;
|
|
case FW_ID_VCE:
|
|
strcpy(name, "VCE");
|
|
break;
|
|
case FW_ID_ISP:
|
|
strcpy(name, "ISP");
|
|
break;
|
|
case FW_ID_DMCU_ERAM:
|
|
strcpy(name, "DMCU_ERAM");
|
|
break;
|
|
case FW_ID_DMCU_ISR:
|
|
strcpy(name, "DMCU_ISR");
|
|
break;
|
|
case FW_ID_RLC_RESTORE_LIST_GPM_MEM:
|
|
strcpy(name, "RLC_RESTORE_LIST_GPM_MEM");
|
|
break;
|
|
case FW_ID_RLC_RESTORE_LIST_SRM_MEM:
|
|
strcpy(name, "RLC_RESTORE_LIST_SRM_MEM");
|
|
break;
|
|
case FW_ID_RLC_RESTORE_LIST_CNTL:
|
|
strcpy(name, "RLC_RESTORE_LIST_CNTL");
|
|
break;
|
|
case FW_ID_RLC_V:
|
|
strcpy(name, "RLC_V");
|
|
break;
|
|
case FW_ID_MMSCH:
|
|
strcpy(name, "MMSCH");
|
|
break;
|
|
case FW_ID_PSP_SYSDRV:
|
|
strcpy(name, "PSP_SYSDRV");
|
|
break;
|
|
case FW_ID_PSP_SOSDRV:
|
|
strcpy(name, "PSP_SOSDRV");
|
|
break;
|
|
case FW_ID_PSP_TOC:
|
|
strcpy(name, "PSP_TOC");
|
|
break;
|
|
case FW_ID_PSP_KEYDB:
|
|
strcpy(name, "PSP_KEYDB");
|
|
break;
|
|
case FW_ID_DFC:
|
|
strcpy(name, "DFC");
|
|
break;
|
|
case FW_ID_PSP_SPL:
|
|
strcpy(name, "PSP_SPL");
|
|
break;
|
|
case FW_ID_DRV_CAP:
|
|
strcpy(name, "DRV_CAP");
|
|
break;
|
|
case FW_ID_MC:
|
|
strcpy(name, "MC");
|
|
break;
|
|
case FW_ID_PSP_BL:
|
|
strcpy(name, "PSP_BL");
|
|
break;
|
|
case FW_ID_CP_PM4:
|
|
strcpy(name, "CP_PM4");
|
|
break;
|
|
case FW_ID_ASD:
|
|
strcpy(name, "ID_ASD");
|
|
break;
|
|
case FW_ID_TA_RAS:
|
|
strcpy(name, "ID_TA_RAS");
|
|
break;
|
|
case FW_ID_XGMI:
|
|
strcpy(name, "ID_XGMI");
|
|
break;
|
|
case FW_ID_RLC_SRLG:
|
|
strcpy(name, "ID_RLC_SRLG");
|
|
break;
|
|
case FW_ID_RLC_SRLS:
|
|
strcpy(name, "ID_RLC_SRLS");
|
|
break;
|
|
case FW_ID_SMC:
|
|
strcpy(name, "ID_SMC");
|
|
break;
|
|
case FW_ID_DMCU:
|
|
strcpy(name, "ID_DMCU");
|
|
break;
|
|
default:
|
|
strcpy(name, "");
|
|
break;
|
|
}
|
|
}
|
|
|
|
|
|
int main() {
|
|
amdsmi_status_t ret;
|
|
|
|
// Init amdsmi for sockets and devices.
|
|
// Here we are only interested in AMD_GPUS.
|
|
ret = amdsmi_init(AMDSMI_INIT_AMD_GPUS);
|
|
CHK_AMDSMI_RET(ret)
|
|
|
|
// Get all sockets
|
|
uint32_t socket_count = 0;
|
|
|
|
// Get the socket count available for the system.
|
|
ret = amdsmi_get_socket_handles(&socket_count, nullptr);
|
|
CHK_AMDSMI_RET(ret)
|
|
|
|
// Allocate the memory for the sockets
|
|
std::vector<amdsmi_socket_handle> sockets(socket_count);
|
|
// Get the sockets of the system
|
|
ret = amdsmi_get_socket_handles(&socket_count, &sockets[0]);
|
|
CHK_AMDSMI_RET(ret)
|
|
|
|
std::cout << "Total Socket: " << socket_count << std::endl;
|
|
|
|
// For each socket, get identifier and devices
|
|
for (uint32_t i = 0; i < socket_count; i++) {
|
|
// Get Socket info
|
|
char socket_info[128];
|
|
ret = amdsmi_get_socket_info(sockets[i], 128, socket_info);
|
|
CHK_AMDSMI_RET(ret)
|
|
std::cout << "Socket " << socket_info << std::endl;
|
|
|
|
// Get the device count available for the socket.
|
|
uint32_t device_count = 0;
|
|
ret = amdsmi_get_processor_handles(sockets[i], &device_count, nullptr);
|
|
CHK_AMDSMI_RET(ret)
|
|
|
|
// Allocate the memory for the device handlers on the socket
|
|
std::vector<amdsmi_processor_handle> processor_handles(device_count);
|
|
// Get all devices of the socket
|
|
ret = amdsmi_get_processor_handles(sockets[i],
|
|
&device_count, &processor_handles[0]);
|
|
CHK_AMDSMI_RET(ret)
|
|
|
|
// For each device of the socket, get name and temperature.
|
|
for (uint32_t j = 0; j < device_count; j++) {
|
|
// Get device type. Since the amdsmi is initialized with
|
|
// AMD_SMI_INIT_AMD_GPUS, the processor_type must be AMD_GPU.
|
|
processor_type_t processor_type = {};
|
|
ret = amdsmi_get_processor_type(processor_handles[j], &processor_type);
|
|
CHK_AMDSMI_RET(ret)
|
|
if (processor_type != AMD_GPU) {
|
|
std::cout << "Expect AMD_GPU device type!\n";
|
|
return AMDSMI_STATUS_NOT_SUPPORTED;
|
|
}
|
|
|
|
// Get BDF info
|
|
amdsmi_bdf_t bdf = {};
|
|
ret = amdsmi_get_gpu_device_bdf(processor_handles[j], &bdf);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf(" Output of amdsmi_get_gpu_device_bdf:\n");
|
|
printf("\tDevice[%d] BDF %04lx:%02x:%02x.%d\n\n", i,
|
|
bdf.domain_number, bdf.bus_number, bdf.device_number,
|
|
bdf.function_number);
|
|
|
|
// Get handle from BDF
|
|
amdsmi_processor_handle dev_handle;
|
|
ret = amdsmi_get_processor_handle_from_bdf(bdf, &dev_handle);
|
|
CHK_AMDSMI_RET(ret)
|
|
|
|
// Get ASIC info
|
|
amdsmi_asic_info_t asic_info = {};
|
|
ret = amdsmi_get_gpu_asic_info(processor_handles[j], &asic_info);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf(" Output of amdsmi_get_gpu_asic_info:\n");
|
|
printf("\tMarket Name: %s\n", asic_info.market_name);
|
|
printf("\tDeviceID: 0x%lx\n", asic_info.device_id);
|
|
printf("\tVendorID: 0x%x\n", asic_info.vendor_id);
|
|
printf("\tRevisionID: 0x%x\n", asic_info.rev_id);
|
|
printf("\tAsic serial: 0x%s\n\n", asic_info.asic_serial);
|
|
|
|
// Get VBIOS info
|
|
amdsmi_vbios_info_t vbios_info = {};
|
|
ret = amdsmi_get_gpu_vbios_info(processor_handles[j], &vbios_info);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf(" Output of amdsmi_get_gpu_vbios_info:\n");
|
|
printf("\tVBios Name: %s\n", vbios_info.name);
|
|
printf("\tBuild Date: %s\n", vbios_info.build_date);
|
|
printf("\tPart Number: %s\n", vbios_info.part_number);
|
|
printf("\tVBios Version String: %s\n\n",
|
|
vbios_info.version);
|
|
|
|
// Get power measure
|
|
amdsmi_power_info_t power_measure = {};
|
|
ret = amdsmi_get_power_info(processor_handles[j], &power_measure);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf(" Output of amdsmi_get_power_info:\n");
|
|
printf("\tCurrent GFX Voltage: %d\n",
|
|
power_measure.gfx_voltage);
|
|
printf("\tAverage socket power: %d\n",
|
|
power_measure.average_socket_power);
|
|
printf("\tGPU Power limit: %d\n\n", power_measure.power_limit);
|
|
|
|
// Get driver version
|
|
char version[AMDSMI_MAX_DRIVER_VERSION_LENGTH];
|
|
int version_length = AMDSMI_MAX_DRIVER_VERSION_LENGTH;
|
|
ret = amdsmi_get_gpu_driver_version(processor_handles[j], &version_length, version);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf(" Output of amdsmi_get_gpu_driver_version:\n");
|
|
printf("\tDriver version: %s\n\n", version);
|
|
|
|
// Get device uuid
|
|
unsigned int uuid_length = AMDSMI_GPU_UUID_SIZE;
|
|
char uuid[AMDSMI_GPU_UUID_SIZE];
|
|
ret = amdsmi_get_gpu_device_uuid(processor_handles[j], &uuid_length, uuid);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf(" Output of amdsmi_get_gpu_device_uuid:\n");
|
|
printf("\tDevice uuid: %s\n\n", uuid);
|
|
|
|
// Get engine usage info
|
|
amdsmi_engine_usage_t engine_usage = {};
|
|
ret = amdsmi_get_gpu_activity(processor_handles[j], &engine_usage);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf(" Output of amdsmi_get_gpu_activity:\n");
|
|
printf("\tAverage GFX Activity: %d\n",
|
|
engine_usage.gfx_activity);
|
|
printf("\tAverage MM Activity: %d\n",
|
|
engine_usage.mm_activity);
|
|
printf("\tAverage UMC Activity: %d\n\n",
|
|
engine_usage.umc_activity);
|
|
|
|
// Get firmware info
|
|
amdsmi_fw_info_t fw_information = {};
|
|
char ucode_name[AMDSMI_MAX_STRING_LENGTH];
|
|
ret = amdsmi_get_fw_info(processor_handles[j], &fw_information);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf(" Output of amdsmi_get_fw_info:\n");
|
|
printf("Number of Microcodes: %d\n", fw_information.num_fw_info);
|
|
for (int j = 0; j < fw_information.num_fw_info; j++) {
|
|
getFWNameFromId(fw_information.fw_info_list[j].fw_id, ucode_name);
|
|
printf(" %s: %ld\n", ucode_name, fw_information.fw_info_list[j].fw_version);
|
|
}
|
|
|
|
// Get GFX clock measurements
|
|
amdsmi_clk_info_t gfx_clk_values = {};
|
|
ret = amdsmi_get_clock_info(processor_handles[j], CLK_TYPE_GFX,
|
|
&gfx_clk_values);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf(" Output of amdsmi_get_clock_info:\n");
|
|
printf("\tGPU GFX Max Clock: %d\n", gfx_clk_values.max_clk);
|
|
printf("\tGPU GFX Current Clock: %d\n", gfx_clk_values.cur_clk);
|
|
|
|
// Get MEM clock measurements
|
|
amdsmi_clk_info_t mem_clk_values = {};
|
|
ret = amdsmi_get_clock_info(processor_handles[j], CLK_TYPE_MEM,
|
|
&mem_clk_values);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf("\tGPU MEM Max Clock: %d\n", mem_clk_values.max_clk);
|
|
printf("\tGPU MEM Current Clock: %d\n\n", mem_clk_values.cur_clk);
|
|
|
|
// Get PCIe status
|
|
amdsmi_pcie_info_t pcie_info = {};
|
|
ret = amdsmi_get_pcie_link_status(processor_handles[j], &pcie_info);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf(" Output of amdsmi_get_pcie_link_status:\n");
|
|
printf("\tPCIe lanes: %d\n", pcie_info.pcie_lanes);
|
|
printf("\tPCIe speed: %d\n\n", pcie_info.pcie_speed);
|
|
|
|
// Get PCIe caps
|
|
amdsmi_pcie_info_t pcie_caps_info = {};
|
|
ret = amdsmi_get_pcie_link_caps(processor_handles[j], &pcie_caps_info);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf(" Output of amdsmi_get_pcie_link_caps:\n");
|
|
printf("\tPCIe max lanes: %d\n", pcie_caps_info.pcie_lanes);
|
|
printf("\tPCIe max speed: %d\n\n", pcie_caps_info.pcie_speed);
|
|
|
|
// Get VRAM temperature limit
|
|
int64_t temperature = 0;
|
|
ret = amdsmi_get_temp_metric(
|
|
processor_handles[j], TEMPERATURE_TYPE_VRAM,
|
|
AMDSMI_TEMP_CRITICAL, &temperature);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf(" Output of amdsmi_get_temp_metric:\n");
|
|
printf("\tGPU VRAM temp limit: %ld\n", temperature);
|
|
|
|
// Get GFX temperature limit
|
|
ret = amdsmi_get_temp_metric(
|
|
processor_handles[j], TEMPERATURE_TYPE_EDGE,
|
|
AMDSMI_TEMP_CRITICAL, &temperature);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf("\tGPU GFX temp limit: %ld\n\n", temperature);
|
|
|
|
// Get temperature measurements
|
|
// amdsmi_temperature_t edge_temp, junction_temp, vram_temp,
|
|
// plx_temp;
|
|
int64_t temp_measurements[4];
|
|
amdsmi_temperature_type_t temp_types[4] = {
|
|
TEMPERATURE_TYPE_EDGE, TEMPERATURE_TYPE_JUNCTION,
|
|
TEMPERATURE_TYPE_VRAM, TEMPERATURE_TYPE_PLX};
|
|
for (const auto &temp_type : temp_types) {
|
|
ret = amdsmi_get_temp_metric(
|
|
processor_handles[j], temp_type,
|
|
AMDSMI_TEMP_CURRENT,
|
|
&temp_measurements[(int)(temp_type)]);
|
|
CHK_AMDSMI_RET(ret)
|
|
}
|
|
printf(" Output of amdsmi_get_temp_metric:\n");
|
|
printf("\tGPU Edge temp measurement: %ld\n",
|
|
temp_measurements[TEMPERATURE_TYPE_EDGE]);
|
|
printf("\tGPU Junction temp measurement: %ld\n",
|
|
temp_measurements[TEMPERATURE_TYPE_JUNCTION]);
|
|
printf("\tGPU VRAM temp measurement: %ld\n",
|
|
temp_measurements[TEMPERATURE_TYPE_VRAM]);
|
|
printf("\tGPU PLX temp measurement: %ld\n\n",
|
|
temp_measurements[TEMPERATURE_TYPE_PLX]);
|
|
|
|
// Get RAS features enabled
|
|
char block_names[14][10] = {"UMC", "SDMA", "GFX", "MMHUB",
|
|
"ATHUB", "PCIE_BIF", "HDP", "XGMI_WAFL",
|
|
"DF", "SMN", "SEM", "MP0",
|
|
"MP1", "FUSE"};
|
|
char status_names[7][10] = {"NONE", "DISABLED", "PARITY",
|
|
"SING_C", "MULT_UC", "POISON",
|
|
"ENABLED"};
|
|
amdsmi_ras_err_state_t state = {};
|
|
int index = 0;
|
|
printf(" Output of amdsmi_get_gpu_ras_block_features_enabled:\n");
|
|
for (auto block = AMDSMI_GPU_BLOCK_FIRST;
|
|
block <= AMDSMI_GPU_BLOCK_LAST;
|
|
block = (amdsmi_gpu_block_t)(block * 2)) {
|
|
ret = amdsmi_get_gpu_ras_block_features_enabled(processor_handles[j], block,
|
|
&state);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf("\tBlock: %s\n", block_names[index]);
|
|
printf("\tStatus: %s\n", status_names[state]);
|
|
index++;
|
|
}
|
|
printf("\n");
|
|
|
|
// Get bad pages
|
|
char bad_page_status_names[3][15] = {"RESERVED", "PENDING",
|
|
"UNRESERVABLE"};
|
|
uint32_t num_pages = 0;
|
|
ret = amdsmi_get_gpu_bad_page_info(processor_handles[j], &num_pages,
|
|
nullptr);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf(" Output of amdsmi_get_gpu_bad_page_info:\n");
|
|
if (!num_pages) {
|
|
printf("\tNo bad pages found.\n");
|
|
} else {
|
|
std::vector<amdsmi_retired_page_record_t> bad_page_info(num_pages);
|
|
ret = amdsmi_get_gpu_bad_page_info(processor_handles[j], &num_pages,
|
|
bad_page_info.data());
|
|
CHK_AMDSMI_RET(ret)
|
|
for (uint32_t page_it = 0; page_it < num_pages; page_it += 1) {
|
|
printf(" Page[%d]\n", page_it);
|
|
printf("\tAddress: %lu\n",
|
|
bad_page_info[page_it].page_address);
|
|
printf("\tSize: %lu\n", bad_page_info[page_it].page_size);
|
|
printf(
|
|
"\tStatus: %s\n",
|
|
bad_page_status_names[bad_page_info[page_it].status]);
|
|
}
|
|
}
|
|
printf("\n");
|
|
|
|
// Get ECC error counts
|
|
amdsmi_error_count_t err_cnt_info = {};
|
|
ret = amdsmi_get_gpu_total_ecc_count(processor_handles[j], &err_cnt_info);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf(" Output of amdsmi_get_gpu_total_ecc_count:\n");
|
|
printf("\tCorrectable errors: %lu\n", err_cnt_info.correctable_count);
|
|
printf("\tUncorrectable errors: %lu\n\n",
|
|
err_cnt_info.uncorrectable_count);
|
|
// Get process list
|
|
auto compare = [](const void *a, const void *b) -> int {
|
|
return (*(amdsmi_proc_info_t *)a).pid >
|
|
(*(amdsmi_proc_info_t *)b).pid
|
|
? 1
|
|
: -1;
|
|
};
|
|
|
|
uint32_t num_process = 0;
|
|
ret = amdsmi_get_gpu_process_list(processor_handles[j], &num_process,
|
|
nullptr);
|
|
CHK_AMDSMI_RET(ret)
|
|
if (!num_process) {
|
|
printf("No processes found.\n");
|
|
} else {
|
|
amdsmi_process_handle_t process_list[num_process];
|
|
amdsmi_proc_info_t info_list[num_process];
|
|
amdsmi_proc_info_t process = {};
|
|
uint64_t mem = 0, gtt_mem = 0, cpu_mem = 0, vram_mem = 0;
|
|
uint64_t gfx = 0, enc = 0;
|
|
char bdf_str[20];
|
|
sprintf(bdf_str, "%04lx:%02x:%02x.%d", bdf.domain_number,
|
|
bdf.bus_number, bdf.device_number, bdf.function_number);
|
|
int num = 0;
|
|
ret = amdsmi_get_gpu_process_list(processor_handles[j], &num_process,
|
|
process_list);
|
|
CHK_AMDSMI_RET(ret)
|
|
for (uint32_t it = 0; it < num_process; it += 1) {
|
|
if (getpid() == process_list[it]) {
|
|
continue;
|
|
}
|
|
ret = amdsmi_get_gpu_process_info(processor_handles[j],
|
|
process_list[it], &process);
|
|
if (ret != AMDSMI_STATUS_SUCCESS) {
|
|
printf("amdsmi_get_gpu_process_info() failed for "
|
|
"process_list[%d], returned %d\n",
|
|
it, ret);
|
|
continue;
|
|
}
|
|
info_list[num++] = process;
|
|
}
|
|
qsort(info_list, num, sizeof(info_list[0]), compare);
|
|
printf("+=======+==================+============+=============="
|
|
"+=============+=============+=============+============"
|
|
"==+=========================================+\n");
|
|
printf(
|
|
"| pid | name | user | gpu bdf | "
|
|
"fb usage | gtt memory | cpu memory | vram memory | "
|
|
"engine usage (ns) |\n");
|
|
printf("| | | | "
|
|
"| | | | "
|
|
" | gfx enc |\n");
|
|
printf("+=======+"
|
|
"+=============+=============+=============+============"
|
|
"==+=========================================+\n");
|
|
for (int it = 0; it < num; it++) {
|
|
char command[30];
|
|
struct passwd *pwd = NULL;
|
|
struct stat st;
|
|
|
|
sprintf(command, "/proc/%d", info_list[it].pid);
|
|
if (stat(command, &st))
|
|
continue;
|
|
pwd = getpwuid(st.st_uid);
|
|
if (!pwd)
|
|
printf("| %5d | %16s | %10d | %s | %7ld KiB | %7ld KiB "
|
|
"| %7ld KiB | %7ld KiB | %lu %lu |\n",
|
|
info_list[it].pid, info_list[it].name, st.st_uid,
|
|
bdf_str, info_list[it].mem / 1024,
|
|
info_list[it].memory_usage.gtt_mem / 1024,
|
|
info_list[it].memory_usage.cpu_mem / 1024,
|
|
info_list[it].memory_usage.vram_mem / 1024,
|
|
info_list[it].engine_usage.gfx,
|
|
info_list[it].engine_usage.enc);
|
|
else
|
|
printf("| %5d | %16s | %10s | %s | %7ld KiB | %7ld KiB "
|
|
"| %7ld KiB | %7ld KiB | %lu %lu |\n",
|
|
info_list[it].pid, info_list[it].name,
|
|
pwd->pw_name, bdf_str, info_list[it].mem / 1024,
|
|
info_list[it].memory_usage.gtt_mem / 1024,
|
|
info_list[it].memory_usage.cpu_mem / 1024,
|
|
info_list[it].memory_usage.vram_mem / 1024,
|
|
info_list[it].engine_usage.gfx,
|
|
info_list[it].engine_usage.enc);
|
|
mem += info_list[it].mem / 1024;
|
|
gtt_mem += info_list[it].memory_usage.gtt_mem / 1024;
|
|
cpu_mem += info_list[it].memory_usage.cpu_mem / 1024;
|
|
vram_mem += info_list[it].memory_usage.vram_mem / 1024;
|
|
gfx = info_list[it].engine_usage.gfx;
|
|
enc = info_list[it].engine_usage.enc;
|
|
printf(
|
|
"+-------+------------------+------------+-------------"
|
|
"-+-------------+-------------+-------------+----------"
|
|
"----+-----------------------------------------+\n");
|
|
}
|
|
printf("| TOTAL:| %s | %7ld "
|
|
"KiB | %7ld KiB | %7ld KiB | %7ld KiB | %lu %lu "
|
|
"%lu %lu %lu |\n",
|
|
bdf_str, mem, gtt_mem, cpu_mem, vram_mem, gfx,
|
|
enc);
|
|
printf("+=======+==================+============+=============="
|
|
"+=============+=============+=============+============"
|
|
"=+==========================================+\n");
|
|
}
|
|
|
|
// Get device name
|
|
amdsmi_board_info_t board_info = {};
|
|
ret = amdsmi_get_gpu_board_info(processor_handles[j], &board_info);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf(" Output of amdsmi_get_gpu_board_info:\n");
|
|
std::cout << "\tdevice [" << j
|
|
<< "]\n\t\tProduct name: " << board_info.product_name
|
|
<< "\n"
|
|
<< "\t\tProduct number: " << board_info.product_serial
|
|
<< "\n"
|
|
<< "\t\tSerial number: " << board_info.serial_number
|
|
<< "\n\n";
|
|
|
|
// Get temperature
|
|
int64_t val_i64 = 0;
|
|
ret = amdsmi_get_temp_metric(processor_handles[j], TEMPERATURE_TYPE_EDGE,
|
|
AMDSMI_TEMP_CURRENT, &val_i64);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf(" Output of amdsmi_get_temp_metric:\n");
|
|
std::cout << "\t\tTemperature: " << val_i64 << "C"
|
|
<< "\n\n";
|
|
|
|
// Get frame buffer
|
|
amdsmi_vram_info_t vram_usage = {};
|
|
ret = amdsmi_get_gpu_vram_usage(processor_handles[j], &vram_usage);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf(" Output of amdsmi_get_gpu_vram_usage:\n");
|
|
std::cout << "\t\tFrame buffer usage (MB): " << vram_usage.vram_used
|
|
<< "/" << vram_usage.vram_total << "\n\n";
|
|
|
|
amdsmi_power_cap_info_t cap_info = {};
|
|
ret = amdsmi_get_power_cap_info(processor_handles[j], 0, &cap_info);
|
|
CHK_AMDSMI_RET(ret)
|
|
printf(" Output of amdsmi_get_power_cap_info:\n");
|
|
std::cout << "\t\t Power Cap: " << cap_info.power_cap
|
|
<< "W\n";
|
|
std::cout << "\t\t Default Power Cap: " << cap_info.default_power_cap
|
|
<< "\n\n";
|
|
std::cout << "\t\t Dpm Cap: " << cap_info.dpm_cap
|
|
<< "\n\n";
|
|
std::cout << "\t\t Min Power Cap: " << cap_info.min_power_cap
|
|
<< "\n\n";
|
|
std::cout << "\t\t Max Power Cap: " << cap_info.max_power_cap
|
|
<< "\n\n";
|
|
}
|
|
}
|
|
|
|
// Clean up resources allocated at amdsmi_init. It will invalidate sockets
|
|
// and devices pointers
|
|
ret = amdsmi_shut_down();
|
|
CHK_AMDSMI_RET(ret)
|
|
|
|
return 0;
|
|
}
|