姚文涛
我是中国科学院计算技术研究所一年级硕士研究生,在华为进行联合培养。我的研究生导师是刘珂教授,华为导师是管紫轩。
我的研究兴趣集中在 computer systems and architecture,目前关注 CXL memory pooling、memory congestion、下一代互连系统、AI accelerators、computer networks 和 LLM KVCache 管理与优化。
代表项目
LocWeave
Locality Isolation for CXL Memory Pooling
CXL memory pooling 可以提升容量和带宽共享能力,但细粒度 interleaving 可能会把不兼容的 requester streams 混入共享内存设备,破坏 memory locality。LocWeave 探索 pattern-aware placement 和 remapping 机制,以降低 multi-host CXL systems 中的干扰。
CXL Memory PoolingMemory CongestionMemory Systemsgem5DRAMSim3
HeteroSim
Memory-Space Design Exploration Simulator for Multi-Chiplet LLM Accelerators
HeteroSim 是一个用于 multi-chiplet LLM accelerators 的 memory-space design exploration simulator。它支持研究 heterogeneous memory organization、chiplet-level resource allocation 和 accelerator data movement 如何影响大模型 inference 与 training 效率。
LLM AcceleratorsMulti-Chiplet SystemsMemory Space DesignArchitecture Simulation
Mooncake Master Snapshot Refactoring
KVCache-Centric Serving System Snapshot Subsystem Refactor
围绕 Mooncake Store 的 RFC #2688,我将 master snapshot 的 orchestration、repository、codec 和 restore path 从 monolithic MasterService 中拆分出来,并通过 4 个已合并 PR 渐进式落地:#2805、#2831、#2879、#2943。该工作保持 snapshot format、配置和外部行为兼容,同时改善模块边界、测试性和后续格式演进空间。
MooncakeKVCache ServingSnapshot/RestoreC++System Refactoring
AI Coprocessor Accelerator for LLM Workloads
Transformer-Oriented Accelerator Design
该项目面向 LLM-oriented Transformer workloads 设计 AI coprocessor accelerator,探索高效 matrix computation、data movement 和 memory access patterns 的硬件支持。
LLM AccelerationAI CoprocessorTransformerHardware Design
Programmable Switch Data-Plane Project
P4-Based Switch and Packet-Processing Design
该项目基于 P4-style programmable data planes 探索 switch behavior 和 packet-processing logic,重点是在 programmable switch 环境中构建并实验网络 forwarding functions。
Programmable SwitchP4Data PlaneComputer Networks
当前方向
- CXL pooling 的 memory systems:理解 shared and disaggregated memory 中的 congestion、interference 和 resource placement。
- 面向 LLMs 的 accelerator architecture:探索 chiplet-level memory organization、data movement 和面向 Transformer workloads 的 AI coprocessor 支持。
- Programmable networking:基于 P4-style data planes 原型化 packet-processing 和 forwarding mechanisms。
- LLM KVCache 系统:研究 KVCache 传输、管理与优化,以及分布式 KVCache serving 系统。
开源贡献
我参与了 Mooncake 开源项目的贡献。Mooncake 是一个面向大模型推理的 KVCache-centric serving 系统,关注高性能 KV cache 传输与分布式 KV cache 管理。
我已向 Mooncake 主分支合并 6 个 Pull Request:#2691、#2754、#2805、#2831、#2879 和 #2943。早期工作主要集中在 Transfer Engine 中的 GPU device 管理与多 GPU NVLink/MNNVL 通信正确性。
近期我围绕 RFC #2688 完成了 Mooncake Store master snapshot 子系统重构,并拆分为四个已合并 PR:
- #2805:提取
MasterSnapshotManager和MasterSnapshotRepository,将 snapshot scheduling、child process lifecycle、storage/catalog operations 从MasterService中拆分。 - #2831:提取
MasterSnapshotCodec,集中管理 master snapshot 的 encode/decode 逻辑,并补充 codec 单元测试。 - #2879:将 snapshot restore path 重构为 repository、codec、service apply 三阶段架构,保持已有 snapshot format 与 restore behavior 兼容。
- #2943:清理重构后的遗留 wrapper methods,并为三阶段 restore 架构补充文档。
