From 46958cf941997264ff36c23c42a71d30674b61f0 Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Sun, 12 Feb 2023 08:15:18 +0000 Subject: [PATCH] BlockAllocator -> BlockManager --- cacheflow/master/block_manager.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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] = {}