Mooncake Transfer Engine 深度解读:从经典 TE 到下一代 TENT
一、前言
Mooncake 是 KVCache.AI 开源的面向大模型推理场景的高性能传输引擎,专门解决 LLM 推理中 KV Cache 在跨节点、跨设备间高效传输的问题。本文基于 Mooncake Transfer Engine 的源码和设计文档,对其核心架构做全面解读。
本文涵盖内容
- Transfer Engine(TE)经典架构概览
- 核心抽象:Batch、Task、Slice
- RDMA 传输的完整实现(RdmaContext、RdmaEndPoint、WorkerPool)
- 多协议支持(MultiTransport)
- 拓扑感知的设备选择(Topology)
- 下一代引擎 TENT(Transfer Engine NEXT)的设计理念和关键技术
二、Transfer Engine(经典 TE)架构概览
2.1 总体设计
Transfer Engine 采用Pimpl(Pointer to Implementation)模式,对外暴露TransferEngine类作为统一入口,具体