ColossalAI/colossalai/pipeline
duanjunwen 1d328ff651 Merge branch 'main' into dev/zero_bubble 2024-11-01 03:10:53 +00:00
..
schedule Merge branch 'main' into dev/zero_bubble 2024-11-01 03:10:53 +00:00
__init__.py [feat] add zerobubble pp (just a frame now); add POC test for dx_dw; add test for zerobubble; 2024-08-22 10:25:34 +00:00
p2p.py [FP8] rebase main (#5963) 2024-08-06 16:29:37 +08:00
stage_manager.py [fix\ fix fail case test_shard_llama 2024-10-25 02:28:55 +00:00
weight_grad_store.py [fix] fix handle name; rm useless comments; 2024-10-29 03:24:15 +00:00