星途科讯 11小时前
AMD MI355X实测:运行Kimi K3性价比碾压英伟达
index_new5.html
../../../zaker_core/zaker_tpl_static/wap/tpl_keji1.html

 

过去数月,开源模型能力呈现爆发式增长。随着 DeepSeek V4-Pro 和 GLM5.2 智能水平逼近 Opus,开源模型已成为闭源模型极具成本效益的替代方案。而 Kimi K3 的出现,更标志着开源新时代的到来。

然而,更强智能往往伴随更大规模。GLM5.2 参数量达 7530 亿,DeepSeek V4-Pro 为 1.6 万亿,Kimi K3 更是高达 2.8 万亿。这意味着在分配 100 万 token 上下文的 KV Cache 前,仅权重加载就需超过 1.5TB 显存。即便是配备 8 张 GPU 的 NVIDIA B200 节点也无法容纳 Kimi K3,部署选择受限:要么使用单卡 288GB 显存的 B300 节点,要么部署两个 B200 节点(TP16)。

AMD MI355X 的性价比优势

在此背景下,拥有 288GB 显存的 AMD MI355X 成为关键变量。其单卡均价约为 B300 的 1/2.4、B200 的 1/1.7,硬件规格对标 Blackwell 架构,极具性价比。尽管 AMD 长期受限于软件生态,但随着其对 Kimi K3 提供首日支持,部署门槛大幅降低。

实测数据显示,在 1,024-token 输入 /400-token 输出的基准测试中,MI355X 节点聚合吞吐量达 952 tok/s,单流解码速度为 118 tok/s。相比双节点 TP16 B200 部署(聚合吞吐量 498 tok/s),MI355X 聚合吞吐量高出 3.8 倍以上,单流速度快 1.3 倍。虽然 B300 节点聚合吞吐量(1,568 tok/s)仍比 MI355X 高约 1.65 倍,但考虑到 B300 价格是 MI355X 的 2.4 倍,MI355X 在每美元性能上实现彻底反超。

值得注意的是,B200 数据偏低部分源于其跨节点全归约开销(RoCE v2),这是唯一需跨越两个节点的配置,因为单节点无法同时容纳 Kimi K3 权重和百万级 KV 池。这也凸显了 MI355X 大显存优势在实际部署中的价值。

技术优化路径

为实现上述性能,团队针对 ROCm 平台进行了两项关键优化:

1. 修复推测解码崩溃

Kimi K3 缺乏内置草稿张量,需依赖外部块扩散草稿。在 ROCm 上,sglang 的接受采样验证器因缺失 top_k_renorm_prob 定义导致调度器崩溃。修复方案并非编写自定义内核,而是直接在 PyTorch 层面实现 Top-k renorm 操作(排序、掩码、重缩放)。此举使单流性能提升约 2.2 倍,峰值聚合吞吐量提升 18%,并支持更高并发。

2. 优化预填充延迟

针对用户感知的 " 首字延迟 ",团队发现 MI355X 在冷预填充阶段耗时远超 B300(51 秒 vs 23 秒)。根源在于 K3 在 TP8 模式下每层 12 个注意力头,与 AITER MLA 内核要求的 4/8/16 倍数不匹配,导致回退至缓慢的 Triton 通用内核。通过将头数零填充至 16,成功调用快速 ASM 内核,使预填充速度提升 2-3 倍,显著改善首字延迟。

结论

在 MI355X 上运行 Kimi K3 的设置相对开箱即用,遇到的框架级 Bug 少于此前 GLM5.2 的部署经验,且无需自定义内核。随着 AMD 对前沿模型支持的完善,其在每美元性能上的优势正逐步转化为实际竞争力,CUDA 的护城河面临严峻挑战。

【星途科讯 图文丨小林 首发于 ZAKER 科技,转载请注明出处】

宙世代

宙世代

ZAKER旗下Web3.0元宇宙平台

一起剪

一起剪

ZAKER旗下免费视频剪辑工具

相关标签

吞吐量 amd 开源 gpu nvidia
相关文章
评论
没有更多评论了
取消

登录后才可以发布评论哦

打开小程序可以发布评论哦

12 我来说两句…
打开 ZAKER 参与讨论