Add get_block_table

4f6f4967 · Woosuk Kwon · 331fa0b0 · 4f6f4967
Commit 4f6f4967 authored Feb 23, 2023 by Woosuk Kwon
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

cacheflow/master/block_manager.py cacheflow/master/block_manager.py +4 -0

No files found.
--- 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]