Add get_block_table

2023-02-23 07:55:14 +00:00 · 2023-02-23 07:55:14 +00:00 · 4f6f4967f6
commit 4f6f4967f6
parent 331fa0b042
1 changed files with 4 additions and 0 deletions
--- 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]