ColossalAI/tests/test_tensor
Ziyue Jiang 6c5996a56e
[Tensor] add module check and bert test (#1031)
* add Embedding

* Add bert test

* polish

* add check module test

* polish

* polish

* polish

* polish
2022-05-26 18:15:42 +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 check and bert test (#1031) 2022-05-26 18:15:42 +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