botbw c54c4fcd15 [hotfix] moe hybrid parallelism benchmark & follow-up fix (#6048) ... * [example] pass use_fp8_comm flag to all plugins * [example] add mixtral benchmark * [moe] refine assertion and check * [moe] fix mixtral & add more tests * [moe] consider checking dp * sp group and moe_dp_group * [mixtral] remove gate tp & add more tests * [deepseek] fix tp & sp for deepseek * [mixtral] minor fix * [deepseek] add deepseek benchmark		2024-09-10 17:30:53 +08:00
..
__init__.py	[shardformer] init shardformer code structure (#3731)	2023-07-04 16:05:01 +08:00
auto_policy.py	[FP8] rebase main (#5963)	2024-08-06 16:29:37 +08:00
base_policy.py	[fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016)	2024-08-22 09:21:34 +08:00
bert.py	[fp8] support hybrid parallel plugin (#5982)	2024-08-12 18:17:05 +08:00
blip2.py	[fp8] support hybrid parallel plugin (#5982)	2024-08-12 18:17:05 +08:00
bloom.py	[fp8] support hybrid parallel plugin (#5982)	2024-08-12 18:17:05 +08:00
chatglm2.py	[Feature] Split cross-entropy computation in SP (#5959)	2024-09-10 12:06:50 +08:00
command.py	[fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016)	2024-08-22 09:21:34 +08:00
deepseek.py	[hotfix] moe hybrid parallelism benchmark & follow-up fix (#6048)	2024-09-10 17:30:53 +08:00
falcon.py	[fp8] support hybrid parallel plugin (#5982)	2024-08-12 18:17:05 +08:00
gpt2.py	[Feature] Split cross-entropy computation in SP (#5959)	2024-09-10 12:06:50 +08:00
gptj.py	[fp8] support hybrid parallel plugin (#5982)	2024-08-12 18:17:05 +08:00
llama.py	[Feature] Split cross-entropy computation in SP (#5959)	2024-09-10 12:06:50 +08:00
mistral.py	[fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016)	2024-08-22 09:21:34 +08:00
mixtral.py	[hotfix] moe hybrid parallelism benchmark & follow-up fix (#6048)	2024-09-10 17:30:53 +08:00
opt.py	[fp8] support hybrid parallel plugin (#5982)	2024-08-12 18:17:05 +08:00
qwen2.py	[Feature] Zigzag Ring attention (#5905)	2024-08-16 13:56:38 +08:00
sam.py	[fp8] support hybrid parallel plugin (#5982)	2024-08-12 18:17:05 +08:00
t5.py	[fp8] support hybrid parallel plugin (#5982)	2024-08-12 18:17:05 +08:00
vit.py	[fp8] support hybrid parallel plugin (#5982)	2024-08-12 18:17:05 +08:00
whisper.py	[fp8] support hybrid parallel plugin (#5982)	2024-08-12 18:17:05 +08:00