[Coloattention] update coloattention tests for checking outputs and gradients#4389

kurisusnowdeng merged 1 commit intohpcaitech:mainhpcaitech/ColossalAI:mainfrom

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