xla

[Distributed] Test if xm.all_reduce produces a single dynamo graph

#4920

Merged