github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-31 11:25:27 +00:00
Code Issues Packages Projects Releases Wiki Activity
05023ecfee
ColossalAI/tests/test_tensor
History
Ziyue Jiang 05023ecfee
[Tensor] TP Linear 1D row (#843)
2022-04-24 13:43:12 +08:00
..
test_tensor_utils [Tensor] TP Linear 1D row (#843) 2022-04-24 13:43:12 +08:00
test_context.py Init Conext supports lazy allocate model memory (#842) 2022-04-22 18:03:35 +08:00
test_linear_tp.py [Tensor] TP Linear 1D row (#843) 2022-04-24 13:43:12 +08:00
test_op.py [hotfix] the bug of numel() in ColoTensor (#845) 2022-04-24 12:32:10 +08:00