[shardformer] sam support flash attention#4316

flybird11111 merged 8 commits into

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

flybird11111:update-sam-flashflybird11111/ColossalAI:update-sam-flashCopy head branch name to clipboard