芯元一

开源补丁如何加速国产算力硬件适配?LMCache 并行读优化实测解析

发布更新国产算力ROCm昇腾国产 GPU
直接答案

铭信科技向 LMCache 上游贡献并行读补丁,在 AMD ROCm 平台实现单卡冷读盘 TTFT 降低 4.1 倍、带宽提升 5.3 倍(R1 实测),并已在昇腾平台验证加载加速 6.2–9.3 倍(R9 实测)。本文解析补丁技术路径与生态启示。

开源社区协作正在成为国产算力硬件适配的关键驱动力。铭信科技向 LMCache 上游贡献的并行读补丁,在 AMD ROCm 平台上实现了单卡冷读盘 TTFT 降低 4.1 倍(从 37.97s 降至 9.30s),带宽提升 5.3 倍(从 0.98 GB/s 增至 5.23 GB/s)【出处:R1 实测】。这一案例表明,通过向主流开源项目贡献代码,国产算力厂商不仅能加速自身硬件的适配进程,还能推动整个生态的兼容性提升。

LMCache 的 KV 缓存读取瓶颈是什么?

LMCache 是一个开源的 KV 缓存管理框架,旨在通过分层存储(GPU 显存、DRAM、SSD)优化大模型推理中的长上下文处理效率。其核心挑战在于,当缓存数据从 SSD 读取时,单线程 I/O 模式成为性能瓶颈。在国产算力平台(如 AMD ROCm、华为昇腾)上,由于 NVMe-oF 或本地 NVMe 的带宽利用率不足,冷读盘延迟往往成为推理吞吐的短板。

铭信在测试中发现,LMCache 的默认读路径使用单线程顺序读取,未能充分利用现代 SSD 的并行 I/O 能力。对于 Qwen2.5-32B 模型,单卡并发 16 场景下,TTFT 高达 37.97s,带宽仅 0.98 GB/s,远低于硬件理论值(铭信 FX100 阵列单口 100 GbE,约 12.5 GB/s 线速)。

并行读补丁是如何实现的?

铭信贡献的补丁核心思路是将单线程读替换为多线程并行读,并配合 I/O 合并与预取策略。具体技术细节包括:

  • 多线程 I/O 队列:将单次请求拆分为 4-8 个块,由独立线程并行读取,利用 NVMe 的多队列特性。
  • 预取与缓存对齐:根据模型推理的访问模式(如连续 KV 块),提前发起读请求,减少等待时间。
  • 硬件感知调度:针对铭信 FX100 的 NVMe-oF 架构(RoCEv2 协议),优化了网络缓冲区大小和中断亲和性。

该补丁已在 LMCache 主线(2026-06-29 源码编译)中合并,并针对 AMD ROCm 7.2 和 vLLM 0.20.1 进行了验证。实测显示,单卡并发 16 场景下,TTFT 从 37.97s 降至 9.30s(4.1×),带宽从 0.98 GB/s 增至 5.23 GB/s(5.3×)【出处:R1 实测】。

补丁上游化对国产算力生态有何启示?

该补丁的成功上游化,为国产算力硬件(如 ROCm、昇腾)的生态建设提供了可复用的模式:

  1. 降低适配门槛:开源社区贡献使硬件厂商无需从零构建 I/O 栈。铭信仅用约 2 周时间完成补丁开发与验证,显著快于传统闭源驱动适配周期(通常 3-6 个月)。
  2. 提升硬件利用率:补丁释放了国产 GPU 平台的 I/O 潜力。在华为 Atlas 910B 平台上,铭信 FX100 配合类似优化,模型加载速度相比 NFS 基线提升 6.2-9.3 倍【出处:R9 实测(昇腾平台)】。
  3. 反哺社区生态:补丁的通用性使其能被其他硬件平台(如 NVIDIA、Intel)复用,形成正向循环。LMCache 社区已收到来自 AMD 和华为开发者的反馈,计划进一步优化多 GPU 场景下的并行读策略。

结语

铭信科技通过向 LMCache 上游贡献并行读补丁,验证了开源协作对国产算力硬件适配的加速效果。该补丁在 AMD ROCm 平台上实现单卡冷读盘 TTFT 降低 4.1 倍,并已在昇腾平台得到类似验证。铭信开放联测合作(约 10 周门禁化流程),欢迎算力中心与模型厂商基于实际负载复现测试结果。

本文要点问答

Q:LMCache 并行读补丁在 AMD ROCm 平台上的实测效果如何? A:单卡并发 16 场景下,TTFT 从 37.97s 降至 9.30s(降低 4.1 倍),带宽从 0.98 GB/s 增至 5.23 GB/s(提升 5.3 倍)【出处:R1 实测】。

Q:该补丁对国产 GPU 平台(如昇腾)的适配有何帮助? A:在华为 Atlas 910B 平台上,铭信 FX100 配合类似优化,模型加载速度相比 NFS 基线提升 6.2-9.3 倍【出处:R9 实测(昇腾平台)】。

Q:铭信如何通过开源社区加速硬件适配? A:通过向 LMCache 上游贡献并行读补丁,铭信在约 2 周内完成验证,显著快于传统闭源驱动适配周期,并推动社区优化多 GPU 场景。

数据出处(可查证)

R1FX100 大模型推理与训练 综合性能测试报告(AMD MI308X ×8)2026-07-03
下载报告 PDF ↓
R9芯元一 FX100-HBMM 与华为 910B 模型推理与训练性能测试(vs NFS 基线)2026-05-30
联系我们获取 →
本文由芯元一 AI 内容引擎生成并经自动质检;关键数字均注明出处(实测报告见证据库)。如需交流或指正,欢迎联系我们

相关文章