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
..
test_hybrid_parallel_grad_clip_norm	[gemini] gemini support extra-dp (#5043)	2023-11-16 21:03:04 +08:00
test_layer	[shardformer] llama support DistCrossEntropy (#5176)	2023-12-13 01:39:14 +08:00
test_model	[pipeline]: fix p2p comm, add metadata cache and support llama interleaved pp (#5134)	2023-12-22 10:44:00 +08:00
__init__.py	[shardformer] adapted T5 and LLaMa test to use kit (#4049)	2023-07-04 16:05:01 +08:00
test_shard_utils.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
test_with_torch_ddp.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00