ColossalAI/tests/test_booster
Frank Lee a9b8402d93
[booster] added the accelerator implementation (#3159)
2023-03-20 13:59:24 +08:00
..
test_mixed_precision [test] fixed torchrec model test (#3167) 2023-03-20 11:40:25 +08:00
test_accelerator.py [booster] added the accelerator implementation (#3159) 2023-03-20 13:59:24 +08:00