[shardformer] update shardformer to use flash attention 2#4392

ver217 merged 2 commits into

hpcaitech:feature/pipelinehpcaitech/ColossalAI:feature/pipelinefrom

flybird11111:flash-attention-2flybird11111/ColossalAI:flash-attention-2Copy head branch name to clipboard