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.

Wenhao Chen 1810b9100f [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		11 months ago
..
test_hybrid_parallel_grad_clip_norm	[gemini] gemini support extra-dp (#5043)	1 year ago
test_layer	[shardformer] llama support DistCrossEntropy (#5176)	12 months ago
test_model	[pipeline]: fix p2p comm, add metadata cache and support llama interleaved pp (#5134)	11 months ago
__init__.py	[shardformer] adapted T5 and LLaMa test to use kit (#4049)	1 year ago
test_shard_utils.py	[misc] update pre-commit and run all files (#4752)	1 year ago
test_with_torch_ddp.py	[misc] update pre-commit and run all files (#4752)	1 year ago