xla

ef8f6c3f - [FSDPv2] Use the global mesh API (#6500)

[FSDPv2] Use the global mesh API (#6500) Summary: Make FSDPv2 to use the global mesh API such that users can just set the global mesh. Test Plan: python test/spmd/test_fsdp_v2.py -v -k test_fsdp_v2_global_mesh