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 |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| colo_parameter.py | ||
| colo_tensor.py | ||
| compute_spec.py | ||
| const.py | ||
| dist_spec_mgr.py | ||
| distspec.py | ||
| op_wrapper.py | ||
| param_op_hook.py | ||
| process_group.py | ||
| shape_consistency.py | ||
| sharding_spec.py | ||
| tensor_spec.py | ||
| utils.py | ||