github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/tests/test_tensor
History
ver217 ad536e308e
[tensor] refactor colo-tensor (#992) 
* refactor colo-tensor and update linear op

* polish code

* polish code

* update ops and unit tests

* update unit tests

* polish code

* rename dist_spec module

* polish code

* polish code

* remove unneeded import

* fix pipelinable
 		2022-05-19 12:44:59 +08:00
..
_utils [tensor] refactor colo-tensor (#992) 2022-05-19 12:44:59 +08:00
test_addmm_tp.py [tensor] refactor colo-tensor (#992) 2022-05-19 12:44:59 +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 colo-tensor (#992) 2022-05-19 12:44:59 +08:00
test_gpt.py [tensor] refactor colo-tensor (#992) 2022-05-19 12:44:59 +08:00
test_graph.py [tensor] refactor colo-tensor (#992) 2022-05-19 12:44:59 +08:00
test_linear_tp.py [tensor] refactor colo-tensor (#992) 2022-05-19 12:44:59 +08:00
test_model.py [tensor] refactor colo-tensor (#992) 2022-05-19 12:44:59 +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