Jiarui Fang 7f76517a85 [Tensor] make a simple net works with 1D row TP (#879)

2022-04-26 18:11:47 +08:00

..

_utils

[tensor]add assert for colo_tensor 1Drow (#846)

2022-04-24 14:12:45 +08:00

test_context.py

colo init context add device attr. (#866)

2022-04-25 14:24:26 +08:00

test_linear_tp.py

[Tensor] Add function to spec and update linear 1Drow and unit tests (#869)

2022-04-26 10:15:26 +08:00

test_net_tp.py

[Tensor] make a simple net works with 1D row TP (#879)

2022-04-26 18:11:47 +08:00

test_op.py

[Tensor] Add some attributes to ColoTensor (#877)

2022-04-26 15:10:47 +08:00

test_tensor.py

[Tensor] Add some attributes to ColoTensor (#877)

2022-04-26 15:10:47 +08:00