NVIDIA与d-Matrix合作：异构计算如何实现10倍AI推理加速

如果你最近在部署 AI 推理服务大概率会遇到一个头疼的问题GPU 资源要么被训练任务占满要么在推理时利用率上不去——尤其是处理长文本、多轮对话这类需要频繁解码的场景时明明模型计算量不大但生成速度就是快不起来。这种“卡脖子”的感觉背后其实是 AI 算力架构的一个深层矛盾通用 GPU 擅长并行计算但在内存带宽敏感的解码任务上往往发挥不出全部实力。而就在最近NVIDIA 与初创公司 d-Matrix 的一项合作似乎正在尝试从基础设施层面破解这个矛盾。根据公开信息双方正在联手打造混合算力基础设施通过将 NVIDIA 的 Hopper/Blackwell GPU 与 d-Matrix 的 Corsair ASIC 芯片组合使用在 Parasail 平台上实现了 10 倍的 Token 生成速率提升。这个数字背后并不是简单的“强强联合”而是一种更务实的思路与其让一款芯片包打天下不如让不同架构的硬件各司其职在推理流水线上形成互补。1. 为什么推理任务正在从“通用”走向“异构”过去几年AI 基础设施的演进主线一直是“更大规模的通用计算”。无论是训练千亿参数模型还是部署推理服务业界首选方案都是堆叠更多 GPU。但这种模式在推理场景下逐渐暴露出两个核心问题。1.1 解码阶段成为推理瓶颈典型的生成式 AI 推理包含两个阶段预填充Prefill和解码Decode。预填充阶段处理整个输入上下文计算密集度高适合 GPU 的并行架构而解码阶段需要逐个生成 Token每次计算量小但频繁访问模型参数对内存带宽极其敏感。当处理长对话或批量请求时解码任务可能占据 70% 以上的推理时间。GPU 虽然算力强大但在这个阶段往往“有力使不出”DRAM 带宽成为实际瓶颈。这就是为什么很多时候你会发现 GPU 利用率显示不高但生成速度依然上不去。1.2 单一架构的能效天花板通用 GPU 的设计目标是在各种场景下保持良好性能这必然意味着在某些特定任务上存在效率折衷。d-Matrix 的 Corsair 芯片采用的数字存内计算D-IMC架构本质上是在芯片层面重构了计算与存储的关系。传统架构需要将数据从内存读取到计算单元处理后再写回内存。而存内计算直接在存储单元旁布置计算逻辑大幅减少了数据搬运开销。对于解码这类需要频繁读取模型权重的任务这种架构能显著降低延迟和功耗。2. 混合算力如何重构推理流水线NVIDIA 与 d-Matrix 的合作方案并不是简单地把两种硬件放在同一个机箱里而是从工作流层面进行了重新设计。2.1 预填充与解码的任务分离在 Parasail 平台上NVIDIA GPU 负责处理计算密集的预填充阶段将用户输入的整个上下文通过注意力机制进行处理生成第一个 Token 所需的隐状态。这个阶段充分利用 GPU 的并行计算优势。随后解码任务被调度到 d-Matrix Corsair ASIC 上执行。Corsair 芯片集成了大容量 SRAM能够将模型权重保持在芯片内部避免频繁访问外部内存。对于解码这种需要反复读取权重、但每次计算量不大的任务这种架构能极大提升吞吐量。2.2 内存带宽的关键作用理解这个方案的价值需要先了解内存带宽对推理性能的影响。下面是一个简化的对比硬件类型计算能力 (TFLOPS)内存带宽 (GB/s)适合任务NVIDIA H100 GPU67 (FP8)3.35TB/s并行计算、预填充d-Matrix Corsair待公开重点优化片内访问序列解码GPU 的高带宽需要服务整个芯片上的所有计算单元而 ASIC 可以针对解码任务优化内存 hierarchy将最频繁访问的数据保留在更靠近计算单元的位置。2.3 混合架构的工程实现挑战这种异构方案在实际部署中面临几个关键技术挑战数据交换效率GPU 处理完预填充后需要将中间状态传递给 ASIC。这个数据传递的延迟和带宽必须足够高效否则会抵消架构优势。负载均衡不同的输入长度和生成长度会导致预填充与解码的任务比例动态变化。系统需要智能地分配资源避免一方空闲另一方拥堵。软件栈统一开发者不希望为不同硬件编写不同代码。混合方案需要提供统一的编程模型让应用无感知地利用异构硬件。3. 从单点优化到基础设施思维这次合作的意义不仅在于技术方案本身更体现了 AI 基础设施建设的思维转变。3.1 从“芯片竞争”到“系统协同”长期以来AI 芯片领域充斥着“取代 GPU”的叙事。但现实是没有任何单一架构能在所有场景下都最优。d-Matrix CEO Sid Sheth 的观点很有代表性“GPU 和专用推理加速器并非竞争对手而是相辅相成。”这种协同思维更符合工程实际。在现有生态基础上引入专用加速器比推倒重来更容易被市场接受。对 NVIDIA 而言拥抱异构计算也意味着从“通用计算供应商”向“系统方案提供商”的演进。3.2 推理基础设施的层次化设计未来的推理基础设施可能会呈现更加清晰的分层结构计算层混合硬件池根据任务特性动态分配资源调度层智能 orchestrator理解工作负载特征并做出最优调度决策服务层统一的 API 和 SDK对应用开发者隐藏底层复杂性这种架构让专用硬件能够“扬长避短”只在最适合的场景下发挥作用而不是被迫处理所有类型的任务。3.3 对开发者的实际影响对于一线开发者和算法工程师这种变化意味着什么性能预测模型需要更新传统的基于 FLOPs 或参数量的性能估算方法会失效需要引入内存访问模式、任务并行度等新维度。部署策略需要调整可能不再需要为峰值负载配置大量 GPU而是采用“GPUASIC”的混合资源池通过动态调度提高整体利用率。优化重点需要转移模型优化不再只关注计算效率还需要考虑内存访问模式、计算图分割点等系统级因素。4. 混合算力的落地路径与挑战虽然混合架构前景诱人但从实验室方案到规模化部署还有很长的路要走。4.1 技术成熟度曲线目前这类方案仍处于早期阶段面临几个关键挑战生态兼容性现有模型和框架大多针对 GPU 优化迁移到异构环境需要大量的适配工作。PyTorch、TensorFlow 等主流框架对专用加速器的支持仍在完善中。成本效益平衡专用芯片需要达到足够大的规模才能摊薄研发成本。在推理负载还没有完全标准化的情况下投资专用硬件存在一定风险。软件栈成熟度异构计算的编程模型、调试工具、性能分析器等配套软件需要时间成熟。开发者体验直接影响 adoption 速度。4.2 可能的发展路径基于当前技术趋势混合算力基础设施可能沿着以下路径演进垂直场景先行在搜索、对话式 AI 等解码密集型场景率先落地验证技术经济性云服务集成通过公有云提供混合推理实例降低用户尝试门槛标准接口确立形成统一的加速器接口标准促进硬件生态多样化编译器技术突破实现工作负载的自动切分和调度减轻开发者负担4.3 对团队的技术准备建议如果你所在团队正在规划 AI 基础设施可以考虑提前布局保持架构开放性避免将系统与特定硬件过度耦合为未来引入异构硬件留出接口。容器化、服务化是基础要求。建立性能评估体系不仅关注峰值算力更要建立基于真实工作负载的评估方法包括延迟、吞吐量、成本等多个维度。关注软件栈演进跟踪 PyTorch、TensorRT、Triton 等框架对异构计算的支持进展在合适的时机进行技术验证。5. 超越技术基础设施的长期价值当我们讨论 AI 芯片时很容易陷入技术参数的比较。但真正重要的是这些技术如何转化为业务价值。5.1 从成本中心到效率引擎传统的 GPU 集群往往被视为“成本中心”——必要的投入但难以持续优化。混合架构提供了一种新的优化思路通过架构创新提升资源利用率将基础设施转化为竞争优势。对于依赖 AI 推理的业务来说响应速度直接影响用户体验。10 倍的 Token 生成速率提升可能意味着从“可用的 AI 服务”到“流畅的 AI 服务”的质变。5.2 推动应用创新当推理成本下降、性能提升后一些之前不经济的应用场景可能变得可行。比如更复杂的多模态推理任务实时性要求极高的交互场景大规模个性化生成应用基础设施的进步往往会催生新的应用范式而不仅仅是优化现有流程。5.3 对行业格局的潜在影响如果混合计算成为主流可能会重塑 AI 芯片行业的竞争格局专用芯片公司有机会在特定细分领域建立优势通用芯片厂商需要加强系统级解决方案能力云服务商可能通过自研芯片第三方芯片的组合打造差异化服务这种多元化生态最终可能让终端用户受益获得更优化、更经济的算力选择。回到我们开头提到的推理服务部署困境。NVIDIA 与 d-Matrix 的合作提示我们解决性能瓶颈不一定需要等待更强大的通用芯片而是可以重新思考计算架构本身。对于一线工程师来说重要的不是追逐最新硬件而是理解不同架构的特性在系统设计阶段就为异构计算留出空间。下一个阶段的 AI 基础设施竞争可能不再是单一芯片的算力比拼而是如何将不同芯片组合成高效系统的能力较量。这种转变需要我们从“芯片思维”升级到“系统思维”而这正是真正的基础设施建设应该有的样子。