duanjunwen a5ec3d4285 [fix] fix mem; use a new model shape; only assert mem less and equal than theo;

2024-09-09 06:38:31 +00:00

..

test_pipeline_utils

[shardformer] refactor pipeline grad ckpt config (#5646)

2024-04-25 15:19:30 +08:00

test_schedule

[fix] fix mem; use a new model shape; only assert mem less and equal than theo;

2024-09-09 06:38:31 +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