CANN SHMEM代码组织结构

代码组织结构【免费下载链接】shmemCANN SHMEM 是面向昇腾平台的多机多卡内存通信库基于OpenSHMEM 标准协议实现跨设备的高效内存访问与数据同步。项目地址: https://gitcode.com/cann/shmemSHMEM组织结构├── 3rdparty // 依赖的第三方库 ├── docs // 文档 ├── examples // 使用样例 ├── include // 头文件 ├── scripts // 相关脚本 ├── src // 源代码 └── tests // 测试用例includeinclude目录下的头文件是按照如下文件层级进行组织的 include/ ├── shmem.h // shmem所有对外API汇总 ├── device/ // device侧头文件 │ ├── shmem_def.h // device侧定义的公共标展接口 │ ├── ub2gm/ // device侧aicore驱动ub2gm数据面低阶接口扩展接口x │ │ ├── shmem_device_rma.h // aicore ub2gm 远端内存访问 RMA 接口 │ │ └── engine/ // aicore 直驱ub2gm 低阶接口扩展接口x │ │ └── shmem_device_mte.h // aicore 直驱 mte接口 │ ├── gm2gm/ // device侧aicore驱动gm2gm数据面高阶和低阶接口扩展接口x │ │ ├── shmem_device_rma.h // aicore high-level RMA 接口 │ │ ├── shmem_device_amo.h // aicore high-level 原子内存操作接口 │ │ ├── shmem_device_so.h // aicore high-level 信号操作接口 │ │ ├── shmem_device_cc.h // aicore high-level 集合通信接口 │ │ ├── shmem_device_p2p_sync.h // aicore high-level p2p同步接口 │ │ ├── shmem_device_mo.h // aicore high-level 内存保序接口 │ │ └── engine/ // aicore 直驱gm2gm 低阶接口扩展接口x │ │ ├── shmem_device_rdma.h // aicore 直驱 rdma 接口 │ │ ├── shmem_device_sdma.h // aicore 直驱 sdma 接口 │ │ └── shmem_device_mte.h // aicore 直驱 mte接口 │ └── team/ // device侧通信域管理接口 │ └── shmem_device_team.h // device侧通信域管理接口 ├── host_device/ // device_host共用目录 │ └── shmem_common_types.h // 公共数据结构等 └── host/ // host侧头文件 ├── shmem_host_def.h // host侧定义的接口 ├── init/ // host侧初始化接口 │ └── shmem_host_init.h // host初始化接口 ├── team/ // host侧通信域管理接口 │ └── shmem_host_team.h // host通信域管理接口 ├── mem/ // host侧内存管理接口 │ └── shmem_host_heap.h // host内存管理接口 ├── data_plane/ // host侧CPU驱动数据面接口 │ ├── shmem_host_rma.h // host high-level RMA 接口 │ ├── shmem_device_so.h // host high-level 信号操作接口 │ ├── shmem_host_cc.h // host high-level 集合通信接口 │ └── shmem_host_p2p_sync.h // host high-level p2p同步接口 └── utils/ // host侧dfx接口 └── shmem_log.h // host dfx-log接口src└── src ├── device // device侧接口实现 └── host ├─bootstrap // bootstrap接口实现 ├─hybm // hybrid memory ├─init // host侧初始化接口实现 ├─mem // host侧内存管理接口实现 ├─python_wrapper // Py接口封装 ├─sync // 同步接口实现 ├─team // host侧通信域管理接口实现 └─transport // 建链相关内容examples└── examples ├── allgather // allgather通信算子样例 ├── allgather_matmul // allgathermatmul融合算子样例 ├── allgather_matmul_padding // 含padding的allgathermatmul融合算子样例 ├── allgather_matmul_with_gather_result // 融合allgather并保留gather的matmul样例 ├── dispatch_gmm_combine // GMM分派与结果合并样例 ├── dynamic_tiling // 动态分块实现样例 ├── kv_shuffle // kv cache shuffle样例 ├── matmul_allreduce // matmulallreduce融合算子样例 ├── matmul_reduce_scatter // matmulreduce_scatter融合算子样例 ├── matmul_reduce_scatter_padding // 含padding的matmulreduce_scatter融合算子样例 ├── rdma_demo // rdma实现样例 ├── rdma_handlewait_test // rdma handle wait测试样例 ├── rdma_perftest // rdma性能测试样例 └── sdma // sdma实现样例tests└── tests ├── examples └── unittest ├── init // 初始化接口单元测试 ├── mem // 内存管理接口单元测试 ├── sync // 同步管理接口单元测试 └── team // 通信域管理接口单元测试docs└── docs ├── api // shmem api ├── debug // shmem debug调测 ├── deployment // SO文件部署与使用指导 ├── doxygen ├── example // 使用样例 ├── images //图片 ├── SECURITY.md //安全声明 ├── code_organization.md // 工程组织架构本文件 ├── compilation_build_guide.md // 相关脚本介绍 ├── conf.py // conf文件 ├── index.rst // 目录树 ├── multi_instance.md // 多实例支持 ├── principles.md // SHMEM原理概述 ├── quickstart.md // 快速开始scripts存放相关脚本。 脚本具体功能和使用【免费下载链接】shmemCANN SHMEM 是面向昇腾平台的多机多卡内存通信库基于OpenSHMEM 标准协议实现跨设备的高效内存访问与数据同步。项目地址: https://gitcode.com/cann/shmem创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考