ver217 c2fdc6a011 [tensor] derive compute pattern from dist spec (#971) ...

* derive compute pattern from dist spec

* polish code

2022-05-16 14:58:08 +08:00

..

__init__.py

add DistSpec for loss and test_model (#947)

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

addmm.py

[tensor] derive compute pattern from dist spec (#971)

2022-05-16 14:58:08 +08:00

element_wise.py

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

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

embedding.py

[tensor] derive compute pattern from dist spec (#971)

2022-05-16 14:58:08 +08:00

layernorm.py

add DistSpec for loss and test_model (#947)

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

linear.py

[tensor] derive compute pattern from dist spec (#971)

2022-05-16 14:58:08 +08:00

loss.py

[tensor] derive compute pattern from dist spec (#971)

2022-05-16 14:58:08 +08:00