[BUG]: Auto parallel case failed in prepare pass phase

[vincentccc]

🐛 Describe the bug

colossalai run --nproc_per_node 2 ColossalAI/examples/language/gpt/experiments/auto_parallel/auto_parallel_with_gpt.py

AttributeError: 'DeviceMesh' object has no attribute 'mesh_shape'

Environment

No response

[vincentccc]

Fix PR:#4502

[flybird11111]

Fix PR:#4502

Thank you for your contribution.