Files
ColossalAI/colossalai/zero/sharded_param/__init__.py
HELSON e5ea3fdeef [gemini] add GeminiMemoryManger (#832)
* refactor StatefulTensor, tensor utilities

* add unitest for GeminiMemoryManager
2022-04-24 13:08:48 +08:00

6 lines
194 B
Python

from colossalai.zero.sharded_param.sharded_tensor import ShardedTensor
from colossalai.zero.sharded_param.sharded_param import ShardedParamV2
__all__ = [
'ShardedTensor', 'ShardedParamV2']