ColossalAI/tests/test_pipeline
Hongxin Liu 1b387ca9fe
[shardformer] refactor pipeline grad ckpt config (#5646)
* [shardformer] refactor pipeline grad ckpt config

* [shardformer] refactor pipeline grad ckpt config

* [pipeline] fix stage manager
2024-04-25 15:19:30 +08:00
..
test_pipeline_utils [shardformer] refactor pipeline grad ckpt config (#5646) 2024-04-25 15:19:30 +08:00
test_schedule [devops] remove post commit ci (#5566) 2024-04-08 15:09:40 +08:00
test_p2p_communication.py Merge branch 'main' into sync/npu 2024-01-18 12:05:21 +08:00
test_stage_manager.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00