ColossalAI/tests/test_tensor/_utils
ver217 ad536e308e
[tensor] refactor colo-tensor (#992)
* refactor colo-tensor and update linear op

* polish code

* polish code

* update ops and unit tests

* update unit tests

* polish code

* rename dist_spec module

* polish code

* polish code

* remove unneeded import

* fix pipelinable
2022-05-19 12:44:59 +08:00
..
__init__.py [tensor]add assert for colo_tensor 1Drow (#846) 2022-04-24 14:12:45 +08:00
_util.py [tensor] refactor colo-tensor (#992) 2022-05-19 12:44:59 +08:00