Sarah Chieng · Cerebras · 阅读原文

原始来源:Sarah Chieng · Cerebras

大模型推理需要存放权重,还要给正在处理的请求保留状态。Cerebras 这篇架构说明展示一种安排:把模型的不同层放在多片晶圆上,权重尽量留在各自的片上 SRAM(直接位于计算芯片上的高速存储器)中,相邻阶段通过网络传递中间计算结果。这样,多片晶圆可以像流水线一样,同时处理处于不同阶段的请求。

文章用一个3万亿参数、93层的 Kimi K3 配置举例。按4位权重计算,权重约需1.5 TB;每片 WSE-3T 有44 GB SRAM,示例还假定平均请求长度为256K、KV缓存使用8位表示。线性注意力层保留的状态与全注意力层不同,后者还要为请求的 KV cache 留空间,即保存此前 token 的键和值,供后续注意力计算复用。因此不能只用总权重除以总存储量,便得出部署方案。

在文章给定的层分配和请求假设下,69个线性注意力层占35片晶圆,24个全注意力层各占一片,合计59片。单个请求依次通过这些阶段;多个请求错开进入,才能让流水线同时工作。微批处理把若干请求一起送入阶段,可改善利用率,也会改变等待时间。

这些数字用于解释容量和调度方式,依赖量化、上下文与状态预算。本篇并未给出可据以独立比较整机吞吐、能耗和实际服务排队的统一测试;“装得下”、单次生成快和在线服务容量是不同问题。

可信度 65 · 重要性 81 · 新颖度 77

技术与主题

系统与效率 · 基础模型 · SRAM · Pipeline Parallelism · KV Cache

zhyblife99

观点原文

文章 HN 讨论中的架构问题,未报告已发生故障。

忠实转述(非原句):他追问:如果一片晶圆出问题,是否会影响整条流水线,以及怎样排空任务和滚动升级。这个问题把注意力从峰值速度引向故障范围和维护方式;本轮已读文章没有给出完整答案。

lifeisloving

观点原文

同一讨论,对代码生成的审阅负担提出异议。

忠实转述(非原句):他认为,在需要认真审阅输出的工作里,生成速度未必是主要瓶颈;更快地产生长篇答案并不自动减少理解和验证的劳动。

8note

观点原文

同一讨论,解释低延迟对个人专注的价值。

忠实转述(非原句):他强调,几秒钟的小等待也会打断注意力。即使之后要花更长时间检查答案,先快速拿到结果仍能保持工作连续性。这是工作方式的理由,不是硬件性能实验。

编辑评读

这里的新能力,是用较大的片上存储和跨晶圆流水线容纳更大的模型;实际价值取决于模型结构与请求分布能否匹配这种安排。全注意力层需要为不同长度的请求留下多少缓存,直接关系到同时能服务多少人。只看一个用户获得 token 的速度,会漏掉进入流水线前的排队,也看不到流量不足时的空闲成本。

读者若在比较推理服务,可以先固定模型、输入输出长度和并发范围,再看首 token 等待、持续生成速度、完成请求数与尾部延迟。用户对快慢的分歧也有实际原因:交互式调试中,短等待可能频繁打断思路;长篇代码生成后,人工审查又可能成为主要耗时。加速前一段仍有价值,但应把省下的等待与新增的有效交付分开记录。本文没有完成这些服务测量,评论里的限流体验也不足以推出硬件上限。

需要留意

  • 59片来自指定模型与状态预算示例;没有本刊独立吞吐、能耗或故障测试。

核查依据

所属事件

收录记录