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
1586d41f
Commit
1586d41f
authored
Sep 25, 2025
by
yangshj1
Browse files
fix connector metadata
parent
c6b7a44b
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
5 additions
and
3 deletions
+5
-3
vllm/v1/core/sched/output.py
vllm/v1/core/sched/output.py
+1
-0
vllm/v1/core/sched/scheduler.py
vllm/v1/core/sched/scheduler.py
+3
-2
vllm/v1/worker/gpu_model_runner.py
vllm/v1/worker/gpu_model_runner.py
+1
-1
No files found.
vllm/v1/core/sched/output.py
View file @
1586d41f
...
...
@@ -155,3 +155,4 @@ class SchedulerOutput:
# KV Cache Connector metadata.
kv_connector_metadata
:
Optional
[
KVConnectorMetadata
]
=
None
kv_lmcache_connector_metadata
:
Optional
[
KVConnectorMetadata
]
=
None
vllm/v1/core/sched/scheduler.py
View file @
1586d41f
...
...
@@ -89,6 +89,7 @@ class Scheduler(SchedulerInterface):
self
.
connector
=
KVConnectorFactory
.
create_connector_v1
(
config
=
self
.
vllm_config
,
role
=
KVConnectorRole
.
SCHEDULER
)
self
.
lmcache_connector
=
None
if
envs
.
VLLM_LMCACHE_ENABLE
:
lmcache_config
=
copy
.
deepcopy
(
self
.
vllm_config
)
lmcache_config
.
kv_transfer_config
=
KVTransferConfig
(
kv_connector
=
"LMCacheConnectorV1"
,
kv_role
=
"kv_both"
)
...
...
@@ -602,7 +603,7 @@ class Scheduler(SchedulerInterface):
if
self
.
lmcache_connector
is
not
None
:
meta
=
self
.
lmcache_connector
.
build_connector_meta
(
scheduler_output
)
scheduler_output
.
kv_connector_metadata
=
meta
scheduler_output
.
kv_
lmcache_
connector_metadata
=
meta
events
=
self
.
kv_cache_manager
.
take_events
()
if
events
:
...
...
@@ -1034,7 +1035,7 @@ class Scheduler(SchedulerInterface):
if
self
.
lmcache_connector
is
not
None
:
meta
=
self
.
lmcache_connector
.
build_connector_meta
(
scheduler_output
)
scheduler_output
.
kv_connector_metadata
=
meta
scheduler_output
.
kv_
lmcache_
connector_metadata
=
meta
events
=
self
.
kv_cache_manager
.
take_events
()
if
events
:
...
...
vllm/v1/worker/gpu_model_runner.py
View file @
1586d41f
...
...
@@ -1747,7 +1747,7 @@ class GPUModelRunner(LoRAModelRunnerMixin):
lmcache_connector
=
get_lmcache_connector
()
if
lmcache_connector
is
not
None
:
lmcache_connector
.
bind_connector_metadata
(
scheduler_output
.
kv_connector_metadata
)
scheduler_output
.
kv_
lmcache_
connector_metadata
)
lmcache_connector
.
start_load_kv
(
get_forward_context
())
@
staticmethod
...
...
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