From b992828ad272dbea38446b1b41c45fb78d049547 Mon Sep 17 00:00:00 2001 From: haowen-han Date: Sun, 15 Feb 2026 16:32:13 +0800 Subject: [PATCH] fix: fix bug on kimi2.5 with dp2 and tp4 (#18604) Co-authored-by: hanhaowen --- python/sglang/srt/layers/attention/trtllm_mla_backend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/sglang/srt/layers/attention/trtllm_mla_backend.py b/python/sglang/srt/layers/attention/trtllm_mla_backend.py index 4f81a46f0..bebe41037 100755 --- a/python/sglang/srt/layers/attention/trtllm_mla_backend.py +++ b/python/sglang/srt/layers/attention/trtllm_mla_backend.py @@ -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],