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.

botbw 696fced0d7 [fp8] fix missing fp8_comm flag in mixtral (#6057)		3 months ago
..
__init__.py	[shardformer] init shardformer code structure (#3731)	1 year ago
auto_policy.py	[FP8] rebase main (#5963)	4 months ago
base_policy.py	[fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016)	3 months ago
bert.py	[fp8] support hybrid parallel plugin (#5982)	4 months ago
blip2.py	[fp8] support hybrid parallel plugin (#5982)	4 months ago
bloom.py	[fp8] support hybrid parallel plugin (#5982)	4 months ago
chatglm2.py	[Feature] Split cross-entropy computation in SP (#5959)	3 months ago
command.py	[fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016)	3 months ago
deepseek.py	[hotfix] moe hybrid parallelism benchmark & follow-up fix (#6048)	3 months ago
falcon.py	[fp8] support hybrid parallel plugin (#5982)	4 months ago
gpt2.py	[Feature] Split cross-entropy computation in SP (#5959)	3 months ago
gptj.py	[fp8] support hybrid parallel plugin (#5982)	4 months ago
llama.py	[Feature] Split cross-entropy computation in SP (#5959)	3 months ago
mistral.py	[fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016)	3 months ago
mixtral.py	[fp8] fix missing fp8_comm flag in mixtral (#6057)	3 months ago
opt.py	[fp8] support hybrid parallel plugin (#5982)	4 months ago
qwen2.py	[Feature] Zigzag Ring attention (#5905)	3 months ago
sam.py	[fp8] support hybrid parallel plugin (#5982)	4 months ago
t5.py	[fp8] support hybrid parallel plugin (#5982)	4 months ago
vit.py	[fp8] support hybrid parallel plugin (#5982)	4 months ago
whisper.py	[fp8] support hybrid parallel plugin (#5982)	4 months ago