From 766c49b17e7397f3609388e12dfcd59dbdd2dfd8 Mon Sep 17 00:00:00 2001 From: Chen Zhang Date: Sat, 10 May 2025 02:28:53 -0700 Subject: [PATCH] rename Signed-off-by: Chen Zhang --- tests/v1/core/test_specialized_manager.py | 2 +- vllm/v1/core/kv_cache_manager.py | 3 ++- ...{specialized_manager.py => single_type_kv_cache_manager.py} | 0 3 files changed, 3 insertions(+), 2 deletions(-) rename vllm/v1/core/{specialized_manager.py => single_type_kv_cache_manager.py} (100%) diff --git a/tests/v1/core/test_specialized_manager.py b/tests/v1/core/test_specialized_manager.py index 540720cb9b2f..de06da0fc11c 100644 --- a/tests/v1/core/test_specialized_manager.py +++ b/tests/v1/core/test_specialized_manager.py @@ -4,7 +4,7 @@ from vllm.v1.core.block_pool import BlockPool from vllm.v1.core.kv_cache_utils import BlockHashType, KVCacheBlock -from vllm.v1.core.specialized_manager import SlidingWindowManager +from vllm.v1.core.single_type_kv_cache_manager import SlidingWindowManager from vllm.v1.kv_cache_interface import SlidingWindowSpec diff --git a/vllm/v1/core/kv_cache_manager.py b/vllm/v1/core/kv_cache_manager.py index c4ed127ece60..ad8468a89dc5 100644 --- a/vllm/v1/core/kv_cache_manager.py +++ b/vllm/v1/core/kv_cache_manager.py @@ -10,7 +10,8 @@ from vllm.v1.core.block_pool import BlockPool from vllm.v1.core.kv_cache_utils import (BlockHashType, KVCacheBlock, hash_request_tokens) -from vllm.v1.core.specialized_manager import get_manager_for_kv_cache_spec +from vllm.v1.core.single_type_kv_cache_manager import ( + get_manager_for_kv_cache_spec) from vllm.v1.kv_cache_interface import KVCacheConfig from vllm.v1.metrics.stats import PrefixCacheStats from vllm.v1.request import Request, RequestStatus diff --git a/vllm/v1/core/specialized_manager.py b/vllm/v1/core/single_type_kv_cache_manager.py similarity index 100% rename from vllm/v1/core/specialized_manager.py rename to vllm/v1/core/single_type_kv_cache_manager.py