From 4f6f4967f6af78534f460d75a9391f9a42b564b0 Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Thu, 23 Feb 2023 07:55:14 +0000 Subject: [PATCH] Add get_block_table --- cacheflow/master/block_manager.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/cacheflow/master/block_manager.py b/cacheflow/master/block_manager.py index 415fe09a..cc3b3849 100644 --- a/cacheflow/master/block_manager.py +++ b/cacheflow/master/block_manager.py @@ -228,3 +228,7 @@ class BlockSpaceManager: for block_table in self.block_tables.values(): self._free_block_table(block_table) self.block_tables.clear() + + def get_block_table(self, seq: Sequence) -> List[int]: + block_table = self.block_tables[seq.seq_id] + return [block.block_number for block in block_table]