Commit 38ed06a8 authored by Woosuk Kwon's avatar Woosuk Kwon
Browse files

Minor

parent 0a11a2e5
...@@ -119,10 +119,10 @@ class BlockSpaceManager: ...@@ -119,10 +119,10 @@ class BlockSpaceManager:
else: else:
# The last block is shared with other sequences. # The last block is shared with other sequences.
# Copy on Write: Allocate a new block and copy the tokens. # Copy on Write: Allocate a new block and copy the tokens.
block = self.gpu_allocator.allocate() new_block = self.gpu_allocator.allocate()
block_table.append(block) block_table.append(new_block)
self.gpu_allocator.free(last_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: def fork(self, src_seq: Sequence, child_seq: Sequence) -> None:
# NOTE: fork does not allocate a new physical block. # NOTE: fork does not allocate a new physical block.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment