ColossalAI/colossalai/zero/shard_utils
HELSON a65cbb7e4e
[zero] refactor shard and gather operation (#773)
2022-04-15 14:41:31 +08:00
..
__init__.py
base_shard_strategy.py
bucket_tensor_shard_strategy.py
commons.py [zero] refactor shard and gather operation (#773) 2022-04-15 14:41:31 +08:00
tensor_shard_strategy.py [zero] refactor shard and gather operation (#773) 2022-04-15 14:41:31 +08:00