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 eee84908d4
[autoparallel] handled illegal sharding strategy (#1728)
2 years ago
..
deprecated [autoparallel] handled illegal sharding strategy (#1728) 2 years ago
node_handler [autoparallel] handled illegal sharding strategy (#1728) 2 years ago
solver [autoparallel] resnet block runtime apply (#1709) 2 years ago
utils [autoparallel] handled illegal sharding strategy (#1728) 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] resnet block runtime apply (#1709) 2 years ago