mindspore-lab / mindformers

Apache License 2.0
144 stars 20 forks source link

Implement input slice for sequence parallel #14

Closed Bourn3z closed 6 months ago

Bourn3z commented 6 months ago

Motivation

Implement input slice for sequence parallel. 为序列并行增加输入切分函数