mirror of https://github.com/hpcaitech/ColossalAI
d73c2b1d79
* change torch.Parameter to ColoParameter * fix post assignment for init context * polish * polish |
||
---|---|---|
.. | ||
_utils | ||
test_addmm_tp.py | ||
test_context.py | ||
test_embedding_tp.py | ||
test_graph.py | ||
test_linear_tp.py | ||
test_model.py | ||
test_op.py | ||
test_tensor.py |