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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-24 22:35:06 +00:00

Files

History

duanjunwen e234dfa236 [feat] support MixtralPipelineForwards--> mixtral_for_causal_lm_forward for zbv

2024-10-10 06:57:35 +00:00

..

test_pipeline_utils

[fix] fix test_pipeline_utils ci;

2024-09-26 07:18:16 +00:00

[feat] support MixtralPipelineForwards--> mixtral_for_causal_lm_forward for zbv

2024-10-10 06:57:35 +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