ColossalAI/tests/test_tensor
Ziyue Jiang d73c2b1d79
[Tensor] fix init context (#931)
* change torch.Parameter to ColoParameter

* fix post assignment for init context

* polish

* polish
2022-05-11 15:48:12 +08:00
..
_utils
test_addmm_tp.py [tensor] hijack addmm for colo tensor (#923) 2022-05-09 18:55:49 +08:00
test_context.py
test_embedding_tp.py update pytest.mark.parametrize in tensor tests (#913) 2022-05-06 11:16:40 +08:00
test_graph.py [Graph] building computing graph with ColoTensor, Linear only (#917) 2022-05-07 17:10:37 +08:00
test_linear_tp.py update pytest.mark.parametrize in tensor tests (#913) 2022-05-06 11:16:40 +08:00
test_model.py [Tensor] fix init context (#931) 2022-05-11 15:48:12 +08:00
test_op.py [Tensor] add from_pretrained support and bert pretrained test (#921) 2022-05-09 16:11:47 +08:00
test_tensor.py