ver217 a3b66f6def [tensor] refactor parallel action (#1007) ...

* refactor parallel action

* polish unit tests

2022-05-20 20:19:58 +08:00

..

__init__.py

add DistSpec for loss and test_model (#947)

2022-05-13 20:29:50 +08:00

_utils.py

[tensor] refactor colo-tensor (#992)

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

addmm.py

[tensor] refactor parallel action (#1007)

2022-05-20 20:19:58 +08:00

element_wise.py

[tensor] refactor colo-tensor (#992)

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

embedding.py

[tensor] refactor parallel action (#1007)

2022-05-20 20:19:58 +08:00

layernorm.py

[tensor] refactor colo-tensor (#992)

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

linear.py

[tensor] refactor parallel action (#1007)

2022-05-20 20:19:58 +08:00

loss.py

[tensor] refactor parallel action (#1007)

2022-05-20 20:19:58 +08:00