From 38ed06a846511e5e048da800a28671519457511c Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Thu, 9 Feb 2023 11:30:55 +0000 Subject: [PATCH] Minor --- 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 70663616d3f5a..0cb02c7544246 100644 --- a/cacheflow/master/block_manager.py +++ b/cacheflow/master/block_manager.py @@ -119,10 +119,10 @@ class BlockSpaceManager: else: # The last block is shared with other sequences. # Copy on Write: Allocate a new block and copy the tokens. - block = self.gpu_allocator.allocate() - block_table.append(block) + new_block = self.gpu_allocator.allocate() + block_table.append(new_block) self.gpu_allocator.free(last_block) - return last_block.block_number, block.block_number + return last_block.block_number, new_block.block_number def fork(self, src_seq: Sequence, child_seq: Sequence) -> None: # NOTE: fork does not allocate a new physical block.