From 5290cef97c26cb46be86f8e6fb5d770d9e581b0f Mon Sep 17 00:00:00 2001 From: Cheng Wan <54331508+ch-wan@users.noreply.github.com> Date: Wed, 17 Dec 2025 12:42:47 -0800 Subject: [PATCH] Fix condition check for require_gathered_buffer (#15328) --- python/sglang/srt/model_executor/model_runner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/sglang/srt/model_executor/model_runner.py b/python/sglang/srt/model_executor/model_runner.py index 861509d9a..a8cc32440 100644 --- a/python/sglang/srt/model_executor/model_runner.py +++ b/python/sglang/srt/model_executor/model_runner.py @@ -2783,7 +2783,7 @@ class ModelRunner: if ( forward_batch.num_token_non_padded is not None and forward_batch.global_num_tokens_gpu is not None - and require_gathered_buffer + and require_gathered_buffer(self.server_args) and not is_nsa_enable_prefill_cp() ): forward_batch.adjust_num_token_non_padded_for_attn_tp(