github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-30 19:05:26 +00:00
Code Issues Packages Projects Releases Wiki Activity
349272c71f
ColossalAI/tests/test_pipeline
History
duanjunwen 349272c71f [fix] updatw bwd b&w input; dict --> list[torch.Tensor]
2024-09-19 07:47:01 +00:00
..
test_pipeline_utils [shardformer] refactor pipeline grad ckpt config (#5646) 2024-04-25 15:19:30 +08:00
test_schedule [fix] updatw bwd b&w input; dict --> list[torch.Tensor] 2024-09-19 07:47:01 +00:00
test_p2p_communication.py [Feature] optimize PP overlap (#5735) 2024-06-26 14:48:02 +08:00
test_stage_manager.py [Feature] optimize PP overlap (#5735) 2024-06-26 14:48:02 +08:00