上周剛寫了 vLLM v0.17.0 的更新，，墨跡未干，v0.17.1 就來了，一個(gè)很小但是很重要的一次更新。

先看全貌：v0.17.1 改了什么

變更

類型

新增 Nemotron 3 Super 模型支持

新模型

[Mamba][Qwen3.5] Zero freed SSM cache blocks on GPU（#35219）

關(guān)鍵修復(fù)

Fix activation_type 傳遞到 TRTLLM fused MoE NVFP4/FP8（#36017）

修復(fù)

恢復(fù) nongated fused moe triton 支持（#36412）

修復(fù)

重新啟用 trtllm MoE FP8 backend 的 EP（#36494）

修復(fù)

Fix TRTLLM Block FP8 MoE Monolithic（#36296）

修復(fù)

[DSV3.2][MTP] 優(yōu)化 Indexer MTP handling（#36723）

? 性能優(yōu)化

重點(diǎn)：35219 修了一個(gè)"越跑越蠢"的 Bug

這個(gè) bug 針對(duì)的是 Qwen3.5-397B-A17B 這類混合架構(gòu)模型。

Qwen3.5 的 MoE 模型用了 Mamba + Attention 的混合架構(gòu)，兩種層共享同一套 GPU block pool。

Mamba 層往 block 里寫的是fp32狀態(tài)數(shù)據(jù)，attention 層用的是fp8或fp16的 KV cache。

問題就出在 block 復(fù)用上

一個(gè) block 先被 Mamba 層用過，留下了fp32的位模式。后來這個(gè) block 被回收，又分配給了 attention 層。attention 層的數(shù)據(jù)類型更窄，fp32殘留的 bit pattern 在新類型下直接變成了NaN或Inf。

最坑的是 attention kernel 的掩碼機(jī)制，很多 attention kernel（FlashAttn3、FlashInfer-TRTLLM 等）處理未使用位置時(shí)，采用的是乘零掩碼——把不需要的位置乘以 0。正常情況下沒問題，但0 × NaN = NaN，這些臟數(shù)據(jù)沿著 KV cache block 一路擴(kuò)散，所有共享這個(gè) block 的請(qǐng)求全部中招。

時(shí)間一長，輸出質(zhì)量持續(xù)下降

這個(gè) Bug 有多隱蔽

看下 Issue #35138 里的復(fù)現(xiàn)條件：

• 模型：Qwen/Qwen3.5-397B-A17B-FP8

• 硬件：8 × NVIDIA B200

• 后端：FlashInfer Attention backend

• 現(xiàn)象：用同一套評(píng)測(cè)腳本跑兩輪，第二輪準(zhǔn)確率就出問題了

重點(diǎn)是第一輪完全正常

服務(wù)剛啟動(dòng)，所有 block 都是干凈的，看什么都好好的。

跑一陣子，block 開始回收和復(fù)用，臟數(shù)據(jù)慢慢滲透，精度悄悄下滑

這種 bug 在生產(chǎn)環(huán)境里排查起來極其痛苦。

模型沒掛、顯存沒炸、API 正常返回，就是結(jié)果在變差。

你可能以為是 prompt 的問題、數(shù)據(jù)的問題、甚至模型本身的問題，但誰能想到是緩存塊復(fù)用帶來的浮點(diǎn)臟數(shù)據(jù)污染。

修復(fù)方案

新分配給 attention 層的 block，用之前先在 GPU 上清零

但做得很克制：

• 只針對(duì)帶 Mamba 層的 hybrid models——純 attention 模型完全不受影響

• 只處理新分配出來的 blocks——prefix cache 命中的不動(dòng)

• 只清 FullAttentionSpec blocks——Mamba 自己的 block 不管，因?yàn)?Mamba 每步都會(huì)完整覆寫狀態(tài)

實(shí)現(xiàn)上用了一個(gè) Triton kernel 批量清零，提前預(yù)計(jì)算好所有 KV cache segment 的絕對(duì)字節(jié)地址，通過 pinned memory 傳 block ID 到 GPU，和 kernel launch 做 overlap，避免同步等待。

性能開銷

官方 PR 給了 B200 上的實(shí)測(cè)數(shù)據(jù)：

階段

清零 blocks 數(shù)

延遲

占 forward step 比例

Prefill（BS ~8K）

~515 blocks（~920 MiB）

~170 μs

Decode

~30 blocks

~15 μs

端到端吞吐測(cè)試，輸出 tokens/s 波動(dòng)在 ±2% 噪聲范圍內(nèi)，代價(jià)幾乎為零

Nemotron 3 Super：這次被寫進(jìn) release notes 了

v0.17.1 另一個(gè)值得注意的變更是新增了Nemotron 3 Super模型支持。

前幾天我剛寫過這個(gè)模型：

總結(jié)

跑 Qwen3.5 混合架構(gòu)模型的：必須升

.5

制作不易，如果這篇文章覺得對(duì)你有用，可否點(diǎn)個(gè)關(guān)注。給我個(gè)三連擊：點(diǎn)贊、轉(zhuǎn)發(fā)和在看。若可以再給我加個(gè)，謝謝你看我的文章，我們下篇再見！