From 2f66b0671b4204d811c828b6e2331a4361fe637b Mon Sep 17 00:00:00 2001 From: Ke Bao Date: Fri, 26 Dec 2025 09:32:35 +0800 Subject: [PATCH] Fix chunk_kda_fwd missing argument (#15851) --- python/sglang/srt/layers/attention/fla/kda.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/python/sglang/srt/layers/attention/fla/kda.py b/python/sglang/srt/layers/attention/fla/kda.py index 357138993..476885746 100644 --- a/python/sglang/srt/layers/attention/fla/kda.py +++ b/python/sglang/srt/layers/attention/fla/kda.py @@ -1147,7 +1147,6 @@ def chunk_kda_fwd( scale: float, initial_state: torch.Tensor, initial_state_indices: torch.Tensor, - output_final_state: bool, cu_seqlens: torch.LongTensor | None = None, ): chunk_size = 64 @@ -1180,7 +1179,6 @@ def chunk_kda_fwd( gk=g, initial_state=initial_state, initial_state_indices=initial_state_indices, - output_final_state=output_final_state, cu_seqlens=cu_seqlens, ) del w, u, kg