Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
55f692b4
Unverified
Commit
55f692b4
authored
Jul 11, 2024
by
Mor Zusman
Committed by
GitHub
Jul 11, 2024
Browse files
[BugFix] get_and_reset only when scheduler outputs are not empty (#6266)
parent
8a1415cf
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
4 deletions
+4
-4
vllm/engine/async_llm_engine.py
vllm/engine/async_llm_engine.py
+2
-2
vllm/engine/llm_engine.py
vllm/engine/llm_engine.py
+2
-2
No files found.
vllm/engine/async_llm_engine.py
View file @
55f692b4
...
...
@@ -225,11 +225,11 @@ class _AsyncLLMEngine(LLMEngine):
"""
seq_group_metadata_list
,
scheduler_outputs
=
self
.
scheduler
[
virtual_engine
].
schedule
()
finished_requests_ids
=
self
.
scheduler
[
virtual_engine
].
get_and_reset_finished_requests_ids
()
if
not
scheduler_outputs
.
is_empty
():
# Execute the model.
finished_requests_ids
=
self
.
scheduler
[
virtual_engine
].
get_and_reset_finished_requests_ids
()
execute_model_req
=
ExecuteModelRequest
(
seq_group_metadata_list
=
seq_group_metadata_list
,
blocks_to_swap_in
=
scheduler_outputs
.
blocks_to_swap_in
,
...
...
vllm/engine/llm_engine.py
View file @
55f692b4
...
...
@@ -871,10 +871,10 @@ class LLMEngine:
"as performance will be severely degraded otherwise."
)
seq_group_metadata_list
,
scheduler_outputs
=
self
.
scheduler
[
0
].
schedule
()
finished_requests_ids
=
self
.
scheduler
[
0
].
get_and_reset_finished_requests_ids
()
if
not
scheduler_outputs
.
is_empty
():
finished_requests_ids
=
self
.
scheduler
[
0
].
get_and_reset_finished_requests_ids
()
execute_model_req
=
ExecuteModelRequest
(
seq_group_metadata_list
=
seq_group_metadata_list
,
blocks_to_swap_in
=
scheduler_outputs
.
blocks_to_swap_in
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment