[shardformer] opt support flash attention#4163

FrankLeeeee merged 6 commits into

hpcaitech:feature/flash-attention-shardformerhpcaitech/ColossalAI:feature/flash-attention-shardformerfrom

flybird11111:feature/flash-attention-shardformerflybird11111/ColossalAI:feature/flash-attention-shardformerCopy head branch name to clipboard