[shardformer] refactored layernorm#4086

FrankLeeeee merged 1 commit intohpcaitech:feature/shardformerhpcaitech/ColossalAI:feature/shardformerfrom

FrankLeeeee:hotfix/layernormFrankLeeeee/ColossalAI:hotfix/layernormCopy head branch name to clipboard