You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_pipeline
Edenzzzz f5c84af0b0
[Feature] Zigzag Ring attention (#5905)
3 months ago
..
test_pipeline_utils [shardformer] refactor pipeline grad ckpt config (#5646) 7 months ago
test_schedule [Feature] Zigzag Ring attention (#5905) 3 months ago
test_p2p_communication.py [Feature] optimize PP overlap (#5735) 5 months ago
test_stage_manager.py [Feature] optimize PP overlap (#5735) 5 months ago