diff --git a/python/sglang/srt/managers/scheduler_pp_mixin.py b/python/sglang/srt/managers/scheduler_pp_mixin.py index 48a34207c..bf088c708 100644 --- a/python/sglang/srt/managers/scheduler_pp_mixin.py +++ b/python/sglang/srt/managers/scheduler_pp_mixin.py @@ -1135,9 +1135,7 @@ class SchedulerPPMixin: batch_result = None send_output_work = [] - all_gather_group = ( - self.attn_tp_group if self.require_attn_tp_allgather else None - ) + all_gather_group = self.attn_tp_group # ---- Prepare send dict ---- # On NPU, always send something (full output or a lightweight skip