[NPU] [bugfix] Fix undefined require_attn_tp_allgather (#39561)

2026-09-16 11:02:40 +08:00
parent 935cbf24ec
commit 5f6dd44edc
1 changed files with 1 additions and 3 deletions
@@ -1135,9 +1135,7 @@ class SchedulerPPMixin:
        batch_result = None
        send_output_work = []
-        all_gather_group = (
+        all_gather_group = self.attn_tp_group
            self.attn_tp_group if self.require_attn_tp_allgather else None
        )
        # ---- Prepare send dict ----
        # On NPU, always send something (full output or a lightweight skip