minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

Fix chunk_kda_fwd missing argument (#15851)

Browse Source
This commit is contained in:
Ke Bao
2025-12-26 09:32:35 +08:00
committed by GitHub
parent 4805118122
commit 2f66b0671b
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/srt/layers/attention/fla/kda.py
-2
View File
@@ -1147,7 +1147,6 @@ def chunk_kda_fwd(
scale: float, scale: float,
initial_state: torch.Tensor, initial_state: torch.Tensor,
initial_state_indices: torch.Tensor, initial_state_indices: torch.Tensor,
output_final_state: bool,
cu_seqlens: torch.LongTensor | None = None, cu_seqlens: torch.LongTensor | None = None,
): ):
chunk_size = 64 chunk_size = 64
@@ -1180,7 +1179,6 @@ def chunk_kda_fwd(
gk=g, gk=g,
initial_state=initial_state, initial_state=initial_state,
initial_state_indices=initial_state_indices, initial_state_indices=initial_state_indices,
output_final_state=output_final_state,
cu_seqlens=cu_seqlens, cu_seqlens=cu_seqlens,
) )
del w, u, kg del w, u, kg