xla
[FSDPv2] Use the global mesh API
#6500
Merged

Loading