[Perf]Optimize waiting queue update with set usage (#20503)

2026-03-19 00:56:24 +08:00
parent 20a23e3173
commit a6c7bb54eb
1 changed files with 2 additions and 3 deletions
--- a/python/sglang/srt/managers/scheduler.py
+++ b/python/sglang/srt/managers/scheduler.py
@@ -2284,9 +2284,8 @@ class Scheduler(
        if len(can_run_list) == 0:
            return None

-        self.waiting_queue = [
-            x for x in self.waiting_queue if x not in set(can_run_list)
-        ]
+        can_run_set = set(can_run_list)
+        self.waiting_queue = [x for x in self.waiting_queue if x not in can_run_set]
        if adder.preempt_list:
            for req in adder.preempt_list:
                self._add_request_to_queue(req)