mirror of https://github.com/hpcaitech/ColossalAI
1c1fe44305
* [autoparallel] adapt solver with self attention * polish code |
||
---|---|---|
.. | ||
deprecated | ||
node_handler | ||
solver | ||
utils | ||
__init__.py | ||
constants.py | ||
sharding_strategy.py |