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
..
__init__.py	[shardformer] adapted T5 and LLaMa test to use kit (#4049)	1 year ago
_utils.py	[pipeline]: fix p2p comm, add metadata cache and support llama interleaved pp (#5134)	11 months ago
test_shard_bert.py	[pipeline]: fix p2p comm, add metadata cache and support llama interleaved pp (#5134)	11 months ago
test_shard_blip2.py	[misc] update pre-commit and run all files (#4752)	1 year ago
test_shard_bloom.py	[hotfix] Add layer norm gradients all-reduce for sequence parallel (#4926)	1 year ago
test_shard_chatglm2.py	[hotfix] Add layer norm gradients all-reduce for sequence parallel (#4926)	1 year ago
test_shard_falcon.py	[shardformer]: support gpt-j, falcon, Mistral and add interleaved pipeline for bert (#5088)	1 year ago
test_shard_gpt2.py	[hotfix] Add layer norm gradients all-reduce for sequence parallel (#4926)	1 year ago
test_shard_gptj.py	[shardformer] llama support DistCrossEntropy (#5176)	12 months ago
test_shard_llama.py	[pipeline]: fix p2p comm, add metadata cache and support llama interleaved pp (#5134)	11 months ago
test_shard_mistral.py	[shardformer]: support gpt-j, falcon, Mistral and add interleaved pipeline for bert (#5088)	1 year ago
test_shard_opt.py	[misc] update pre-commit and run all files (#4752)	1 year ago
test_shard_sam.py	[misc] update pre-commit and run all files (#4752)	1 year ago
test_shard_t5.py	[misc] update pre-commit and run all files (#4752)	1 year ago
test_shard_vit.py	[hotfix] fix torch 2.0 compatibility (#4936)	1 year ago
test_shard_whisper.py	[misc] update pre-commit and run all files (#4752)	1 year ago