Skip to content

附录 B:核心符号速查 ​

符号版本职责定义位置章节
BufferV1Python 用户 API 与通信路径选择deep_ep/buffer.py03、04
deep_ep::BufferV1资源持有、IPC/NVSHMEM 生命周期与 kernel 接口csrc/deep_ep.hpp, csrc/deep_ep.cpp03
get_dispatch_layoutV1从 top-k 生成 rank/expert 计数与 membershipdeep_ep/buffer.py, csrc/kernels/layout.cu04
EPBufferV2EP 主 API、容量参数、handle 与 stream 控制deep_ep/buffers/ep.py05 至 09
EPHandleV2保存 dispatch 布局与 combine 所需元数据deep_ep/buffers/ep.py07、08
BufferBaseV2各专用 buffer 的共同生命周期入口deep_ep/buffers/base.py10
ContextV2NCCL device communicator、domain 与 memory windowcsrc/kernels/comm/api.hpp, context.cpp06
launch_dispatchV2dispatch specialization 与 kernel launchcsrc/kernels/ep/dispatch.hpp05、07
launch_combineV2combine specialization 与 kernel launchcsrc/kernels/ep/combine.hpp05、08
EventOverlapV2封装 CUDA event、Tensor 生命周期和延后回调deep_ep/utils/event.py08
BufferAllocatorV2.5规划对称通信 storage 上的张量布局deep_ep/buffers/allocator.py09、10