You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_tensor
ver217 ad536e308e
[tensor] refactor colo-tensor (#992)
3 years ago
..
_utils [tensor] refactor colo-tensor (#992) 3 years ago
test_addmm_tp.py [tensor] refactor colo-tensor (#992) 3 years ago
test_context.py [tensor] refactor colo-tensor (#992) 3 years ago
test_dist_spec_mgr.py [tensor] refactor colo-tensor (#992) 3 years ago
test_embedding_tp.py [tensor] refactor colo-tensor (#992) 3 years ago
test_gpt.py [tensor] refactor colo-tensor (#992) 3 years ago
test_graph.py [tensor] refactor colo-tensor (#992) 3 years ago
test_linear_tp.py [tensor] refactor colo-tensor (#992) 3 years ago
test_model.py [tensor] refactor colo-tensor (#992) 3 years ago
test_op.py [tensor] refactor colo-tensor (#992) 3 years ago
test_tensor.py [tensor] refactor colo-tensor (#992) 3 years ago