You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/auto_parallel/tensor_shard
Frank Lee 22a115406b
[autoparallel] fixed broken node handler tests (#1708)
2 years ago
..
deprecated [autoparallel] collated all deprecated files (#1700) 2 years ago
node_handler [autoparallel] fixed broken node handler tests (#1708) 2 years ago
solver [autoparallel] refactored the autoparallel module for organization (#1706) 2 years ago
utils [autoparallel] refactored the autoparallel module for organization (#1706) 2 years ago
__init__.py [autoparallel] init new folder structure (#1696) 2 years ago
constants.py [autoparallel] refactored the autoparallel module for organization (#1706) 2 years ago
sharding_strategy.py [autoparallel] refactored the autoparallel module for organization (#1706) 2 years ago