def _storage_size()

in diffq/torch_pack.py [0:0]

• 11 lines of code
• 5 McCabe index (conditional complexity)

def _storage_size(dtype: torch.dtype):
    if dtype == torch.int64:
        return 64
    elif dtype == torch.int32:
        return 32
    elif dtype == torch.int16:
        return 16
    elif dtype == torch.uint8:
        return 8
    else:
        raise ValueError("Invalid bitpacking storage type")