ColossalAI/colossalai/zero/shard_utils at ff4f5d7231d2f2e7387cd6175649fc678aaf6f39 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-11 09:24:09 +00:00

Files

History

HELSON 7c079d9c33 [hotfix] fixed bugs in ShardStrategy and PcieProfiler (#394 )

2022-03-11 18:12:46 +08:00

..

__init__.py

[zero] a shard strategy in granularity of tensor (#307 )

2022-03-11 15:50:28 +08:00

base_shard_strategy.py

[zero] update zero context init with the updated test utils (#327 )

2022-03-11 15:50:28 +08:00

tensor_shard_strategy.py

[hotfix] fixed bugs in ShardStrategy and PcieProfiler (#394 )

2022-03-11 18:12:46 +08:00