ColossalAI/tests/test_pipeline at c114d1429af8f029fa73d0253bb8d07756c99f80 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-17 02:00:25 +00:00

Files

History

duanjunwen 1739df423c [fix] fix fwd branch, fwd pass both micro_batch & internal_inputs'

2024-09-20 07:34:43 +00:00

..

test_pipeline_utils

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

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

[fix] fix fwd branch, fwd pass both micro_batch & internal_inputs'

2024-09-20 07:34:43 +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