mirror of https://github.com/hpcaitech/ColossalAI
49216d7ab1
* [autoparallel] fix bugs caused by negative dim key * fix import error * fix matmul test issue * fix unit test issue |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| colo_parameter.py | ||
| colo_tensor.py | ||
| comm_spec.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 | ||