ColossalAI/colossalai/pipeline
duanjunwen cfade4c36d [feat] Linear1D_COL/ROW support zbv WeightGradStore; 2024-10-14 07:02:43 +00:00
..
schedule [feat] Linear1D_COL/ROW support zbv WeightGradStore; 2024-10-14 07:02:43 +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 fix object_to_tensor usage when torch>=2.3.0 (#5820) 2024-07-16 13:59:25 +08:00
stage_manager.py [plugin] hybrid support zero bubble pipeline (#6060) 2024-09-27 14:48:55 +08:00
weight_grad_store.py [feat] Linear1D_COL/ROW support zbv WeightGradStore; 2024-10-14 07:02:43 +00:00