xla

08574298 - [Distributed] Test if xm.all_reduce produces a single dynamo graph (#4920)

[Distributed] Test if xm.all_reduce produces a single dynamo graph (#4920) Summary: Test if xm.all_reduce produces a single dynamo graph. Test Plan: PJRT_DEVICE=TPU python test/test_mp_replication.py