From 66face3598f25fb4980cd0523b759da2f9ea60cb Mon Sep 17 00:00:00 2001 From: YanbingJiang Date: Sun, 28 Sep 2025 15:31:08 +0800 Subject: [PATCH] Fix CI failure of TypeError: RotaryEmbedding.forward_cpu() got an unexpected keyword argument 'fused_set_kv_buffer_arg' (#11009) --- python/sglang/srt/layers/rotary_embedding.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/sglang/srt/layers/rotary_embedding.py b/python/sglang/srt/layers/rotary_embedding.py index fd69ea727..cc4e21b58 100644 --- a/python/sglang/srt/layers/rotary_embedding.py +++ b/python/sglang/srt/layers/rotary_embedding.py @@ -206,6 +206,7 @@ class RotaryEmbedding(CustomOp): query: torch.Tensor, key: torch.Tensor, offsets: Optional[torch.Tensor] = None, + fused_set_kv_buffer_arg=None, ) -> Tuple[torch.Tensor, torch.Tensor]: positions = torch.add(positions, offsets) if offsets is not None else positions if _is_cpu_amx_available: