7 Commits (a799ca343b13665661a5e95f5ad1523457bef2e2)

Author	SHA1	Message	Date
Wenhao Chen	bb0a668fee	[hotfix] set return_outputs=False in examples and polish code (#5404) ... * fix: simplify merge_batch * fix: use return_outputs=False to eliminate extra memory consumption * feat: add return_outputs warning * style: remove `return_outputs=False` as it is the default value	2024-03-25 12:31:09 +08:00
Wenhao Chen	ef4f0ee854	[hotfix]: add pp sanity check and fix mbs arg (#5268) ... * fix: fix misleading mbs arg * feat: add pp sanity check * fix: fix 1f1b sanity check	2024-01-15 15:57:40 +08:00
Wenhao Chen	4fa689fca1	[pipeline]: fix p2p comm, add metadata cache and support llama interleaved pp (#5134) ... * test: add more p2p tests * fix: remove send_forward_recv_forward as p2p op list need to use the same group * fix: make send and receive atomic * feat: update P2PComm fn * feat: add metadata cache in 1f1b * feat: add metadata cache in interleaved pp * feat: modify is_xx_stage fn * revert: add _broadcast_object_list * feat: add interleaved pp in llama policy * feat: set NCCL_BUFFSIZE in HybridParallelPlugin	2023-12-22 10:44:00 +08:00
Hongxin Liu	079bf3cb26	[misc] update pre-commit and run all files (#4752) ... * [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format	2023-09-19 14:20:26 +08:00
Baizhou Zhang	660eed9124	[pipeline] set optimizer to optional in execute_pipeline (#4630) ... * set optimizer to optional in execute_pipeline * arrange device and mixed precision in booster init * fix execute_pipeline in booster.py	2023-09-07 10:42:59 +08:00
Hongxin Liu	508ca36fe3	[pipeline] 1f1b schedule receive microbatch size (#4589)	2023-09-01 21:45:14 +08:00
LuGY	d3c6cd66f3	[pipeline] add unit test for 1f1b (#4303) ... * add unit test for 1f1b * polish code * polish code and update ut version * fix	2023-08-15 23:25:14 +08:00