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

fix import error (#880)

2022-04-26 19:28:40 +08:00

colo_init_context.py

[tensor] refactor colo-tensor (#992)

2022-05-19 12:44:59 +08:00

pipelinable.py

[tensor] refactor colo-tensor (#992)

2022-05-19 12:44:59 +08:00

utils.py

[pipelinable]use pipelinable context to initialize non-pipeline model (#816)

2022-04-24 13:03:12 +08:00