使用什么库或者SDK可以获取amd的GPU使用率和GPU内存使用率信息,使用C++代码

时间: 2024-05-10 19:13:45 浏览: 17
您可以使用AMD的ROCm SDK提供的API获取AMD GPU的使用率和内存使用率信息。具体来说,您可以使用Power Profiling API获取GPU使用率,使用Performance Counter API获取GPU内存使用率。 下面是使用C代码获取AMD GPU使用率和内存使用率的示例: 获取GPU使用率: ``` #include <roctracer_ext.h> void get_gpu_usage(int device_id) { uint32_t domain = ACTIVITY_DOMAIN_HSA_AMD_GPU_ID; // activity domain for AMD GPU roctracer_properties_t props; memset(&props, 0, sizeof(roctracer_properties_t)); props.buffer_size = 1024; props.buffer_callback_fun = [](uint32_t cid, const void* buffer, size_t size, uint32_t source_id) { roctracer_record_t* record = (roctracer_record_t*)buffer; for (int i = 0; i < size / sizeof(roctracer_record_t); i++) { if (record[i].domain == ACTIVITY_DOMAIN_HSA_AMD_GPU_ID && record[i].kind == ACTIVITY_KIND_HSA_AMD_GPUPROFILE) { roctracer_activity_hsa_amd_gpu_profile_t* prof = (roctracer_activity_hsa_amd_gpu_profile_t*)(record[i].payload); if (prof->dev_index == device_id) { printf("GPU usage: %.2f%%\n", (float)prof->data.curutil / 100.0f); } } } }; roctracer_enable_callback_ex(1, &domain, 1, &props); // do something to trigger GPU activities roctracer_disable_callback_ex(); } ``` 其中,`device_id`参数是AMD GPU的设备ID,可以通过AMD的ROCm API获取。函数中使用了`roctracer`库提供的回调函数机制,监听AMD GPU的`ACTIVITY_KIND_HSA_AMD_GPUPROFILE`事件并解析其payload,从而获取GPU使用率信息。 获取GPU内存使用率: ``` #include <hsa.h> #include <hsa_ext_amd.h> void get_gpu_mem_usage(int device_id) { hsa_amd_memory_pool_t* pool; hsa_amd_memory_pool_global_flag_t flags = (hsa_amd_memory_pool_global_flag_t)( HSA_AMD_MEMORY_POOL_GLOBAL_FLAG_FINE_GRAINED | HSA_AMD_MEMORY_POOL_GLOBAL_FLAG_COARSE_GRAINED); hsa_amd_agent_memory_pool_get_info_fn_t mem_pool_info_fn = nullptr; hsa_amd_agent_memory_pool_get_info_fn_t* mem_pool_info_fn_ptr = &mem_pool_info_fn; hsa_agent_t agent; hsa_status_t status; status = hsa_iterate_agents([](hsa_agent_t agent, void* data) -> hsa_status_t { hsa_device_type_t device_type; hsa_status_t status = hsa_agent_get_info(agent, HSA_AGENT_INFO_DEVICE, &device_type); if (status == HSA_STATUS_SUCCESS && device_type == HSA_DEVICE_TYPE_GPU) { hsa_agent_t* gpu_agent = reinterpret_cast<hsa_agent_t*>(data); *gpu_agent = agent; return HSA_STATUS_INFO_BREAK; } return HSA_STATUS_SUCCESS; }, &agent); status = hsa_amd_agent_memory_pool_get_info(agent, HSA_AMD_AGENT_MEMORY_POOL_INFO_NUM, &mem_pool_info_fn_ptr); uint32_t num_pools = mem_pool_info_fn(agent, HSA_AMD_AGENT_MEMORY_POOL_INFO_NUM); for (int i = 0; i < num_pools; i++) { pool = reinterpret_cast<hsa_amd_memory_pool_t*>(mem_pool_info_fn(agent, HSA_AMD_AGENT_MEMORY_POOL_INFO_PTR + i)); hsa_amd_memory_pool_global_flag_t pool_flags; status = hsa_amd_memory_pool_get_info(*pool, HSA_AMD_MEMORY_POOL_INFO_GLOBAL_FLAGS, &pool_flags); if ((flags & pool_flags) == flags) { // fine-grained + coarse-grained hsa_amd_memory_pool_info_t info; hsa_amd_memory_pool_get_info(*pool, HSA_AMD_MEMORY_POOL_INFO_ACCESSIBLE_BY_ALL, &info); if (info == false) { // if the pool is not accessible by all agents, skip it continue; } hsa_amd_memory_pool_statistics_t stats; hsa_amd_memory_pool_get_info(*pool, HSA_AMD_MEMORY_POOL_INFO_STATISTICS, &stats); if (stats.size == 0) { // if the pool has no memory usage statistics, skip it continue; } hsa_amd_memory_pool_usage_t* usages = new hsa_amd_memory_pool_usage_t[stats.size]; hsa_amd_memory_pool_get_info(*pool, HSA_AMD_MEMORY_POOL_INFO_USAGE, usages); for (int j = 0; j < stats.size; j++) { if (usages[j].agent == agent && usages[j].size > 0) { // the memory usage of GPU printf("GPU memory usage: %.2f%%\n", (float)usages[j].size * 100.0f / stats.max_size); } } delete[] usages; } } } ``` 函数中使用了AMD的HSA API获取AMD GPU的内存池信息,然后从中筛选出可用的内存池,并获取其内存使用率信息。其中,`device_id`参数无用,因为HSA API只会获取当前运行的GPU的信息。函数中需要额外注意内存池是否可被所有Agent访问的问题。 以上代码仅供参考,具体实现方式可能会因环境和需求不同而有所差异。

相关推荐

最新推荐

recommend-type

微信小程序 使用腾讯地图SDK详解及实现步骤

主要介绍了微信小程序 使用腾讯地图SDK详解及实现步骤的相关资料,需要的朋友可以参考下
recommend-type

Qt(C++)调用工业相机Basler的SDK使用示例

主要介绍了Qt(C++)调用工业相机Basler的SDK使用示例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
recommend-type

详解OkSocket与Android的简单使用

本篇文章主要介绍了详解OkSocket与Android的简单使用,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧
recommend-type

Python使用微信SDK实现的微信支付功能示例

主要介绍了Python使用微信SDK实现的微信支付功能,结合实例形式分析了Python调用微信SDK接口实现微信支付功能的具体步骤与相关操作技巧,需要的朋友可以参考下
recommend-type

科大讯飞sdk使用说明文档

科大讯飞sdk使用说明文档,详细介绍了科大讯飞语音识别模块的使用方式和接口定义
recommend-type

zigbee-cluster-library-specification

最新的zigbee-cluster-library-specification说明文档。
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

MATLAB结构体与对象编程:构建面向对象的应用程序,提升代码可维护性和可扩展性

![MATLAB结构体与对象编程:构建面向对象的应用程序,提升代码可维护性和可扩展性](https://picx.zhimg.com/80/v2-8132d9acfebe1c248865e24dc5445720_1440w.webp?source=1def8aca) # 1. MATLAB结构体基础** MATLAB结构体是一种数据结构,用于存储和组织相关数据。它由一系列域组成,每个域都有一个名称和一个值。结构体提供了对数据的灵活访问和管理,使其成为组织和处理复杂数据集的理想选择。 MATLAB中创建结构体非常简单,使用struct函数即可。例如: ```matlab myStruct
recommend-type

详细描述一下STM32F103C8T6怎么与DHT11连接

STM32F103C8T6可以通过单总线协议与DHT11连接。连接步骤如下: 1. 将DHT11的VCC引脚连接到STM32F103C8T6的5V电源引脚; 2. 将DHT11的GND引脚连接到STM32F103C8T6的GND引脚; 3. 将DHT11的DATA引脚连接到STM32F103C8T6的GPIO引脚,可以选择任一GPIO引脚,需要在程序中配置; 4. 在程序中初始化GPIO引脚,将其设为输出模式,并输出高电平,持续至少18ms,以激活DHT11; 5. 将GPIO引脚设为输入模式,等待DHT11响应,DHT11会先输出一个80us的低电平,然后输出一个80us的高电平,
recommend-type

JSBSim Reference Manual

JSBSim参考手册,其中包含JSBSim简介,JSBSim配置文件xml的编写语法,编程手册以及一些应用实例等。其中有部分内容还没有写完,估计有生之年很难看到完整版了,但是内容还是很有参考价值的。