[Scheduler] Unify per-iteration request intake into ingest_requests() (#38389)
This commit is contained in:
@@ -114,8 +114,7 @@ class SchedulerMultiplexMixin:
|
||||
while True:
|
||||
with torch.cuda.stream(decode_stream):
|
||||
set_pdmux_status(False)
|
||||
recv_reqs = self.request_receiver.recv_requests()
|
||||
self.process_input_requests(recv_reqs)
|
||||
self.ingest_requests()
|
||||
running_batch = self.running_batch
|
||||
|
||||
with torch.cuda.stream(prefill_stream):
|
||||
|
||||
Reference in New Issue
Block a user