ROCT进程间共享内存实现分析

最新推荐文章于 2023-07-11 22:42:03 发布

denglin12315

最新推荐文章于 2023-07-11 22:42:03 发布

阅读量706

点赞数

文章标签： ROCT share memory

本文链接：https://blog.csdn.net/denglin12315/article/details/123551226

版权

本文深入探讨了ROCT中进程间共享内存的实现，重点在于shareHandle的使用。export进程通过hsaKmtAllocMemroy、hsaKmtMapMemoryToGPU等API申请和共享内存，而import进程则通过hsaKmtRegisterSharedHandle进行导入。关键在于shareHandle，它在export进程中创建并与内存handle关联，在import进程中用于查找物理内存。共享share_handle可借助Linux标准库的shmget、shmat等函数实现。

摘要由CSDN通过智能技术生成

一.export进程涉及的API

(1) hsaKmtAllocMemroy——申请一块内存对应的虚拟地址和物理地址

(2) hsaKmtMapMemoryToGPU——为虚拟地址和物理地址建立映射关系

(3) hsaKmtShareMemory——将申请的内存共享出去(创建shareHandle，通过参数返回)

(4) hsaKmtUnmapMemoryToGPU——解除虚拟地址和物理内存的映射

(5) hsaKmtFreeMemory——释放分配的虚拟地址和物理内存

二.import进程涉及的API

(1) hsaKmtRegisterSharedHandle——导入export进程创建的共享内存(通过参数传入shareHandle)，并为其分配虚拟地址

(2) hsaKmtMapMemoryToGPU——为虚拟地址和共享内存建立映射关系

(3) hsaKmtUnmapMemoryToGPU —— 解除虚拟地址和共享内存的映射

(4) hsaKmtDeregisterMemory —— 释放分配的虚拟地址

三.可见ROCT进程间内存共享实现的关键是shareHandle

1.shareHandle是什么？

struct kfd_ioctl_ipc_export_handle_args {
	__u64 handle;		/* to KFD */
	__u32 share_handle[4];	/* from KFD */
	__u32 gpu_id;		/* to KFD */
	__u32 flags;		/* to KF