[SPMD] Introduce global mesh (#6498)
Summary:
SPMD expects the mesh to be the same across the board. Therefore, introduce the concept of a global mesh to reduce the need to carry the same mesh in the code.
Test Plan:
python test/spmd/test_xla_sharding.py -v -k test_global_mesh
