diff --git a/cacheflow/master/block_manager.py b/cacheflow/master/block_manager.py index 0cb02c7544246..ab41ae389f72c 100644 --- a/cacheflow/master/block_manager.py +++ b/cacheflow/master/block_manager.py @@ -7,7 +7,7 @@ from cacheflow.sequence import SequenceStatus from cacheflow.utils import Device -class BlockAllocator: +class BlockManager: def __init__( self, @@ -63,8 +63,8 @@ class BlockSpaceManager: self.num_total_gpu_blocks = num_gpu_blocks self.num_total_cpu_blocks = num_cpu_blocks - self.gpu_allocator = BlockAllocator(Device.GPU, block_size, num_gpu_blocks) - self.cpu_allocator = BlockAllocator(Device.CPU, block_size, num_cpu_blocks) + self.gpu_allocator = BlockManager(Device.GPU, block_size, num_gpu_blocks) + self.cpu_allocator = BlockManager(Device.CPU, block_size, num_cpu_blocks) # Mapping: seq_id -> BlockTable. self.block_tables: Dict[int, BlockTable] = {}