Support block size 32 (#35)

2023-04-09 23:07:18 -07:00
parent ee88a7e5f3
commit b9926f7f66
4 changed files with 49 additions and 5 deletions
--- a/cacheflow/master/block_manager.py
+++ b/cacheflow/master/block_manager.py
@@ -15,9 +15,9 @@ class BlockAllocator:
        block_size: int,
        num_blocks: int,
    ) -> None:
-        if block_size not in [8, 16]:
+        if block_size not in [8, 16, 32]:
            raise ValueError(f'Unsupported block size: {block_size}'
-                             'The block size must be either 8 or 16.')
+                             'The block size must be one of {8, 16, 32}.')
        self.device = device
        self.block_size = block_size
        self.num_blocks = num_blocks