ColossalAI/tests/test_tensor
Ziyue Jiang 26d4ab8b03
[Tensor] Add function to spec and update linear 1Drow and unit tests (#869)
2022-04-26 10:15:26 +08:00
..
_utils
test_context.py colo init context add device attr. (#866) 2022-04-25 14:24:26 +08:00
test_linear_tp.py [Tensor] Add function to spec and update linear 1Drow and unit tests (#869) 2022-04-26 10:15:26 +08:00
test_net_tp.py [tensor] add cross_entrophy_loss (#868) 2022-04-25 16:01:52 +08:00
test_op.py