[test] fixed tests failed due to dtensor change#4082

FrankLeeeee merged 2 commits into

hpcaitech:feature/shardformerhpcaitech/ColossalAI:feature/shardformerfrom

FrankLeeeee:hotfix/unit-testFrankLeeeee/ColossalAI:hotfix/unit-testCopy head branch name to clipboard