Skip to content

Commit 8651af6

Browse files
committed
rename file and remove mooncake cruft
Signed-off-by: NickLucche <nlucches@redhat.com>
1 parent 79100d7 commit 8651af6

5 files changed

Lines changed: 3 additions & 7 deletions

File tree

tests/v1/kv_connector/unit/test_nixl_connector_hma.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ def test_read_blocks_for_req_expands_remote_ids(
128128
from vllm.distributed.kv_transfer.kv_connector.v1.nixl.metadata import (
129129
NixlConnectorMetadata,
130130
)
131-
from vllm.distributed.kv_transfer.kv_connector.v1.nixl.transfer_plan import (
131+
from vllm.distributed.kv_transfer.kv_connector.v1.nixl.tp_mapping import (
132132
TPMapping,
133133
)
134134
from vllm.distributed.kv_transfer.kv_connector.v1.nixl.worker import (

tests/v1/kv_connector/unit/test_transfer_plan.py renamed to tests/v1/kv_connector/unit/test_tp_mapping.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111

1212
import pytest
1313

14-
from vllm.distributed.kv_transfer.kv_connector.v1.nixl.transfer_plan import (
14+
from vllm.distributed.kv_transfer.kv_connector.v1.nixl.tp_mapping import (
1515
TPMapping,
1616
compute_tp_mapping,
1717
)

vllm/distributed/kv_transfer/kv_connector/v1/mooncake/mooncake_connector.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -821,7 +821,6 @@ def __init__(
821821
self.cache_config = vllm_config.cache_config
822822
self.kv_cache_config = kv_cache_config
823823
self.use_mla = self.model_config.use_mla
824-
self._physical_blocks_per_logical_kv_block = 1
825824
self._sync_block_size_with_kernel()
826825

827826
# Get the attention backend from the first layer
@@ -864,9 +863,6 @@ def _sync_block_size_with_kernel(self) -> None:
864863
kernel_block_size,
865864
)
866865
assert self.block_size > kernel_block_size
867-
self._physical_blocks_per_logical_kv_block = (
868-
self.block_size // kernel_block_size
869-
)
870866
self.block_size = kernel_block_size
871867

872868
def __del__(self):

vllm/distributed/kv_transfer/kv_connector/v1/nixl/transfer_plan.py renamed to vllm/distributed/kv_transfer/kv_connector/v1/nixl/tp_mapping.py

File renamed without changes.

vllm/distributed/kv_transfer/kv_connector/v1/nixl/worker.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@
4444
from vllm.distributed.kv_transfer.kv_connector.v1.nixl.stats import (
4545
NixlKVConnectorStats,
4646
)
47-
from vllm.distributed.kv_transfer.kv_connector.v1.nixl.transfer_plan import (
47+
from vllm.distributed.kv_transfer.kv_connector.v1.nixl.tp_mapping import (
4848
ReadSpec,
4949
TPMapping,
5050
_is_attention_spec,

0 commit comments

Comments
 (0)