xla

[Distributed] Make xm.all_gather a single graph in Dynamo

#4922

Merged