mirror of https://github.com/hpcaitech/ColossalAI
c4b1b65931
* [test] fixed tests failed due to dtensor change * polish code |
||
|---|---|---|
| .. | ||
| common_utils | ||
| core | ||
| model | ||
| test_dtensor | ||
| test_colo_checkpoint_tools.py | ||
| test_comm_spec_apply.py | ||
| test_context.py | ||
| test_mix_gather.py | ||
| test_parameter.py | ||
| test_shape_consistency.py | ||
| test_shape_consistency_apply.py | ||
| test_sharded_linear.py | ||
| test_sharding_spec.py | ||
| test_tp_with_zero.py | ||