[Bugfix][PD] Fix DSV4 disaggregated serving (#45831)

Signed-off-by: ZhanqiuHu <zhu@redhat.com>
This commit is contained in:
zhanqiuhu
2026-06-17 15:17:14 +00:00
committed by GitHub
parent 46f74e144b
commit eb0fdeb1e8
@@ -71,6 +71,7 @@ from vllm.v1.kv_cache_interface import (
FullAttentionSpec,
MambaSpec,
MLAAttentionSpec,
SlidingWindowMLASpec,
UniformTypeKVCacheSpecs,
)
from vllm.v1.worker.block_table import BlockTable
@@ -962,7 +963,9 @@ class NixlBaseConnectorWorker:
)
else:
self.block_len_per_layer.append(physical_page_size)
is_mla_region = isinstance(layer_spec, MLAAttentionSpec)
is_mla_region = isinstance(
layer_spec, (MLAAttentionSpec, SlidingWindowMLASpec)
)
self._region_is_mla.append(is_mla_region)
if not is_mla_region: