ver217 45b9124df4 [tensor] hijack addmm for colo tensor (#923) ...

* hijack addmm for colo tensor

* fix bugs

* polish unit test

* polish comments

2022-05-09 18:55:49 +08:00

..

__init__.py

[tensor] hijack addmm for colo tensor (#923)

2022-05-09 18:55:49 +08:00

addmm.py

[tensor] hijack addmm for colo tensor (#923)

2022-05-09 18:55:49 +08:00

element_wise.py

[Tensor] make ColoTensor more robust for getattr (#886)

2022-04-27 10:57:49 +08:00

embedding.py

[Tensor] add embedding tp1d row (#904)

2022-04-29 14:10:05 +08:00

layernorm.py

[Tensor] add ColoTensor TP1Dcol Embedding (#899)

2022-04-28 17:45:06 +08:00

linear.py

[Graph] building computing graph with ColoTensor, Linear only (#917)

2022-05-07 17:10:37 +08:00

loss.py

[Tensor] add 1d vocab loss (#918)

2022-05-07 15:49:14 +08:00