fix: fix bug on kimi2.5 with dp2 and tp4 (#18604)

Co-authored-by: hanhaowen <hanhaowen@baidu.com>
2026-02-15 16:32:13 +08:00
parent 274bf6607a
commit b992828ad2
1 changed files with 1 additions and 1 deletions
@@ -1198,7 +1198,7 @@ class TRTLLMMLABackend(FlashInferMLAAttnBackend):
                out=out,
            )
        else:
-            out = torch.empty(
+            out = torch.zeros(
                q.shape[0],
                q.shape[1],
                v.shape[2],