Unverified Commit 55f692b4 authored by Mor Zusman's avatar Mor Zusman Committed by GitHub
Browse files

[BugFix] get_and_reset only when scheduler outputs are not empty (#6266)

parent 8a1415cf
...@@ -225,11 +225,11 @@ class _AsyncLLMEngine(LLMEngine): ...@@ -225,11 +225,11 @@ class _AsyncLLMEngine(LLMEngine):
""" """
seq_group_metadata_list, scheduler_outputs = self.scheduler[ seq_group_metadata_list, scheduler_outputs = self.scheduler[
virtual_engine].schedule() virtual_engine].schedule()
finished_requests_ids = self.scheduler[
virtual_engine].get_and_reset_finished_requests_ids()
if not scheduler_outputs.is_empty(): if not scheduler_outputs.is_empty():
# Execute the model. # Execute the model.
finished_requests_ids = self.scheduler[
virtual_engine].get_and_reset_finished_requests_ids()
execute_model_req = ExecuteModelRequest( execute_model_req = ExecuteModelRequest(
seq_group_metadata_list=seq_group_metadata_list, seq_group_metadata_list=seq_group_metadata_list,
blocks_to_swap_in=scheduler_outputs.blocks_to_swap_in, blocks_to_swap_in=scheduler_outputs.blocks_to_swap_in,
......
...@@ -871,10 +871,10 @@ class LLMEngine: ...@@ -871,10 +871,10 @@ class LLMEngine:
"as performance will be severely degraded otherwise.") "as performance will be severely degraded otherwise.")
seq_group_metadata_list, scheduler_outputs = self.scheduler[ seq_group_metadata_list, scheduler_outputs = self.scheduler[
0].schedule() 0].schedule()
finished_requests_ids = self.scheduler[
0].get_and_reset_finished_requests_ids()
if not scheduler_outputs.is_empty(): if not scheduler_outputs.is_empty():
finished_requests_ids = self.scheduler[
0].get_and_reset_finished_requests_ids()
execute_model_req = ExecuteModelRequest( execute_model_req = ExecuteModelRequest(
seq_group_metadata_list=seq_group_metadata_list, seq_group_metadata_list=seq_group_metadata_list,
blocks_to_swap_in=scheduler_outputs.blocks_to_swap_in, blocks_to_swap_in=scheduler_outputs.blocks_to_swap_in,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment