註冊並分享邀請連結,可獲得影片播放與邀請獎勵。

Byron Hsu
@hsu_byron
Inference/RL @periodiclabs | @lmsysorg ex-@xAI
加入 November 2017
2K 正在關注    6.1K 粉絲
Scaling Kimi K3 on H200s with @sgl_project @lmsysorg While most discussion on X focuses on Blackwell, I thought it would be useful to share what we learned while tuning Kimi K3 for throughput-oriented workloads on H200s. - TP+EP beat DP+EP for K3. Only about 1/4 of its layers use gated MLA, so DP attention provides less benefit from avoiding KV-cache duplication. Having each DP rank process every attention head also substantially increased latency. This differs from our K2.6 and GLM setups, where we generally prefer DP+EP. - FP8 KV cache worked well. We measured no quality or throughput regression in our evaluations, while effectively doubling KV-cache capacity. - TP32+EP32 provided significantly more cache capacity: approximately 4M KV-cache tokens per replica, compared with about 800K using TP16+EP16. - For low cache-hit workloads, consider disabling radix cache. Prefix caching requires four Mamba-state slots per sequence in our setup; without radix cache, each sequence needs only one. - Non-speculative decoding won at high concurrency. Above roughly 64 concurrent requests, draft-and-verify overhead outweighed the savings from accepted speculative tokens in our tests. - A few caveats: symmetric memory caused instability in our multi-node environment. We also observed hangs when combining speculative decoding, HiCache, and high KV pressure. Both issues are still under investigation. We’ve been running this configuration at full load across roughly ~O(1000) chips for several days. Aside from hardware failures, it has been reliable. Feedback, discussion, and better configurations are very welcome! Sharing my launch command in the first comment. h/t to @hanming_lu @liin1211 @0xishand @ChengWan17 @NealBayya for the discussion
顯示更多
0
10
378
30
轉發到社區