[coloattention] coloattention support flash attention 2#4347

kurisusnowdeng merged 1 commit into

hpcaitech:mainhpcaitech/ColossalAI:mainfrom

flybird11111:update-coloattentionflybird11111/ColossalAI:update-coloattentionCopy head branch name to clipboard