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/colossalai/pipeline
duanjunwen 1d328ff651
Merge branch 'main' into dev/zero_bubble
4 weeks ago
..
schedule Merge branch 'main' into dev/zero_bubble 4 weeks ago
__init__.py [feat] add zerobubble pp (just a frame now); add POC test for dx_dw; add test for zerobubble; 3 months ago
p2p.py [FP8] rebase main (#5963) 4 months ago
stage_manager.py [fix\ fix fail case test_shard_llama 1 month ago
weight_grad_store.py [fix] fix handle name; rm useless comments; 4 weeks ago