[Fix] Apply the attention-CP broadcast result in PP dynamic-chunk profiling (#37669)

2026-09-02 17:20:07 -07:00
parent 3421d4375b
commit 5c46ce37f5
4 changed files with 30 additions and 67 deletions
@@ -103,8 +103,8 @@ class TestPPCPRankOffsets(unittest.TestCase):
                side_effect=fake_point_to_point_pyobj,
            ),
            patch(
-                "sglang.srt.managers.scheduler_pp_mixin.broadcast_pyobj",
-                side_effect=lambda data, *args, **kwargs: data,
+                "sglang.srt.managers.scheduler_pp_mixin.attn_cp_tp_broadcast_pyobj",
+                side_effect=lambda data: data,
            ),
        ):
            self.assertEqual(