Support various block sizes & Change default block size to 16 (#38)

2023-04-15 09:03:24 -07:00
parent 84eee24e20
commit 0f4b32199e
7 changed files with 594 additions and 611 deletions
--- a/cacheflow/master/block_manager.py
+++ b/cacheflow/master/block_manager.py
@@ -15,9 +15,6 @@ class BlockAllocator:
        block_size: int,
        num_blocks: int,
    ) -> None:
-        if block_size not in [8, 16, 32]:
-            raise ValueError(f'Unsupported block size: {block_size}'
-                             'The block size must be one of {8, 16, 32}.')
        self.device = device
        self.block_size = block_size
        self.num_blocks = num_blocks