Skip to content

Commit a53ad62

Browse files
authored
[KV offload][1b/N] rename offloading to kv_offload (#25191)
Signed-off-by: Or Ozeri <oro@il.ibm.com>
1 parent 1c3dad2 commit a53ad62

File tree

5 files changed

+5
-5
lines changed

5 files changed

+5
-5
lines changed

.buildkite/test-pipeline.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -280,7 +280,7 @@ steps:
280280
# split the test to avoid interference
281281
- pytest -v -s v1/core
282282
- pytest -v -s v1/executor
283-
- pytest -v -s v1/offloading
283+
- pytest -v -s v1/kv_offload
284284
- pytest -v -s v1/sample
285285
- pytest -v -s v1/logits_processors
286286
- pytest -v -s v1/worker

tests/v1/offloading/test_worker.py renamed to tests/v1/kv_offload/test_worker.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
# SPDX-License-Identifier: Apache-2.0
22
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3-
from vllm.v1.offloading.abstract import LoadStoreSpec
4-
from vllm.v1.offloading.worker.worker import (OffloadingHandler,
3+
from vllm.v1.kv_offload.abstract import LoadStoreSpec
4+
from vllm.v1.kv_offload.worker.worker import (OffloadingHandler,
55
OffloadingWorker, TransferResult,
66
TransferSpec)
77

File renamed without changes.

vllm/v1/offloading/mediums.py renamed to vllm/v1/kv_offload/mediums.py

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

55
import numpy as np
66

7-
from vllm.v1.offloading.abstract import LoadStoreSpec
7+
from vllm.v1.kv_offload.abstract import LoadStoreSpec
88

99

1010
class BlockIDsLoadStoreSpec(LoadStoreSpec, ABC):

vllm/v1/offloading/worker/worker.py renamed to vllm/v1/kv_offload/worker/worker.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
from abc import ABC, abstractmethod
44

55
from vllm.logger import init_logger
6-
from vllm.v1.offloading.abstract import LoadStoreSpec
6+
from vllm.v1.kv_offload.abstract import LoadStoreSpec
77

88
# a single transfer spec (src_blocks_spec, dst_blocks_spec)
99
TransferSpec = tuple[LoadStoreSpec, LoadStoreSpec]

0 commit comments

Comments
 (0)