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
78aa341d
Unverified
Commit
78aa341d
authored
May 14, 2025
by
Russell Bryant
Committed by
GitHub
May 14, 2025
Browse files
[CI] Fix race condition in test_kv_cache_events test (#18169)
Signed-off-by:
Russell Bryant
<
rbryant@redhat.com
>
parent
79747367
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
1 addition
and
2 deletions
+1
-2
tests/v1/engine/test_engine_core_client.py
tests/v1/engine/test_engine_core_client.py
+0
-1
vllm/distributed/kv_events.py
vllm/distributed/kv_events.py
+1
-1
No files found.
tests/v1/engine/test_engine_core_client.py
View file @
78aa341d
...
...
@@ -292,7 +292,6 @@ def test_kv_cache_events(
log_stats
=
False
,
)
endpoint
=
publisher_config
.
endpoint
.
replace
(
"*"
,
"127.0.0.1"
)
time
.
sleep
(
0.1
)
subscriber
=
MockSubscriber
(
endpoint
,
topic
=
publisher_config
.
topic
,
decode_type
=
KVEventBatch
)
...
...
vllm/distributed/kv_events.py
View file @
78aa341d
...
...
@@ -130,6 +130,7 @@ class ZmqEventPublisher(EventPublisher):
self
.
_endpoint
=
endpoint
self
.
_replay_endpoint
=
replay_endpoint
self
.
_hwm
=
hwm
self
.
_socket_setup
()
# Payload
self
.
_seq_gen
=
count
()
...
...
@@ -207,7 +208,6 @@ class ZmqEventPublisher(EventPublisher):
def
_publisher_thread
(
self
)
->
None
:
"""Background thread that processes the event queue."""
self
.
_pack
=
msgspec
.
msgpack
.
Encoder
()
self
.
_socket_setup
()
assert
self
.
_pub
is
not
None
# narrows type for mypy
...
...
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