Class MemoryBucket

checkpoint_engine/pin_memory.py:288–290

    bucket_size = max(4 << 30, max(_align_size(x.dtype, x.shape) for x in parameters.values()))

    class MemoryBucket(BaseModel):
        size: int
        metas: list[ParameterMeta]

    buckets: list[MemoryBucket] = []
    buckets.append(MemoryBucket(size=0, metas=[]))