Function patch_align

scripts/benchmark_gct_memory.py:110–117

(value: int, patch: int, mode: str)

def patch_align(value: int, patch: int, mode: str) -> int:
    if value % patch == 0:
        return value
    if mode == "error":
        raise ValueError(f"{value} is not divisible by patch size {patch}")
    if mode == "floor":
        return (value // patch) * patch
    return ((value + patch - 1) // patch) * patch


def resolve_dtype(device: str, dtype: str) -> torch.dtype: