Skip to content

[shardformer] Support 4D parallel + Flash Attention#5789

Merged
Edenzzzz merged 12 commits intohpcaitech:mainfrom
Edenzzzz:sp_flash_attn
Jun 17, 2024
Merged

[shardformer] Support 4D parallel + Flash Attention#5789
Edenzzzz merged 12 commits intohpcaitech:mainfrom
Edenzzzz:sp_flash_attn

Commits

Commits on Jun 6, 2024

Commits on Jun 13, 2024

Commits on Jun 14, 2024

Commits on Jun 15, 2024

Commits on Jun 16, 2024

Commits on Jun 17, 2024