Ziyue Jiang 32291dd73f [Tensor] add module handler for linear (#1021) ... * add module spec for linear * polish * polish * polish		2022-05-26 11:50:44 +08:00
..
_utils	[unit test] refactor test tensor (#1005)	2022-05-19 18:57:56 +08:00
test_addmm_tp.py	[tensor] refactor parallel action (#1007)	2022-05-20 20:19:58 +08:00
test_context.py	[tensor] refactor colo-tensor (#992)	2022-05-19 12:44:59 +08:00
test_dist_spec_mgr.py	[tensor] refactor colo-tensor (#992)	2022-05-19 12:44:59 +08:00
test_embedding_tp.py	[tensor] refactor parallel action (#1007)	2022-05-20 20:19:58 +08:00
test_gpt.py	[tensor] impl ColoDDP for ColoTensor (#1009)	2022-05-21 13:52:04 +08:00
test_graph.py	[tensor] refactor colo-tensor (#992)	2022-05-19 12:44:59 +08:00
test_linear_tp.py	[tensor] refactor parallel action (#1007)	2022-05-20 20:19:58 +08:00
test_model.py	[tensor] refactor parallel action (#1007)	2022-05-20 20:19:58 +08:00
test_module_spec.py	[Tensor] add module handler for linear (#1021)	2022-05-26 11:50:44 +08:00
test_op.py	[tensor] refactor colo-tensor (#992)	2022-05-19 12:44:59 +08:00
test_tensor.py	[tensor] refactor colo-tensor (#992)	2022-05-19 12:44:59 +08:00