3
运行时
DCU是否不支持ROCM_SMI_LIB相关API?
2025-04-30 11:08:24
123

bool HIP_get_pid_mem() {
processInfos.clear();

if (!g_rsmi_initialized && !init_rsmi()) {
return false;
}

uint32_t num_items = 0;

rsmi_status_t ret = rsmi_compute_process_info_get(nullptr, &num_items);
if (ret != RSMI_STATUS_SUCCESS && ret != RSMI_STATUS_INSUFFICIENT_SIZE) {
std::cerr << "Failed to get process count: " << ret << std::endl;
return false;
}

if (num_items == 0) {
return true;
}

rsmi_process_info_t* procs = new rsmi_process_info_t[num_items];
ret = rsmi_compute_process_info_get(procs, &num_items);
if (ret != RSMI_STATUS_SUCCESS) {
std::cerr << "Failed to get process information: " << ret << std::endl;
delete[] procs;
return false;
}

for (uint32_t i = 0; i < num_items; i++) {
HipProcessInfo info;
info.pid = procs[i].process_id;
info.usedHIPMemory = procs[i].vram_usage;
processInfos.push_back(info);
std::cout << "Process ID: " << info.pid
<< ", Used HIP Memory: " << info.usedHIPMemory / (1024*1024)
<< " MiB" << std::endl;
}

delete[] procs;
// rsmi_shut_down();

return true;
}
当我使用HIP_get_pid_mem()这个函数在DCU上获取显存使用量的时候,我发现该函数给我返回了一个超大的结果:
"Process ID: 3506246, Used HIP Memory: 133496575 MiB"
请问这是为什么?

全部评论(3)
只看楼主
X
xingh
1楼
您好,不支持的,dtk提供的接口只有hip_runtime_api.h这里面的
2025/4/30
网
网友zceNt9
回复 @xingh
那hip(DCU)是否有办法通过C/C++接口获取到DCU显存使用量?还是说只能基于rocm-smi?
2025/4/30
X
xingh
回复 @xingh
hipMemGetInfo这个接口
2025/4/30
没有更多内容了
--
楼主
Level.2
0 帖子
楼主热帖
暂无数据