mirror of https://github.com/hpcaitech/ColossalAI
702dbc5288
* [tensor] use communication autograd func * change all to all comm spec info * rename pattern and distinguish fwd/bwd * polish code |
||
---|---|---|
.. | ||
solver | ||
__init__.py |