xla

[Backport]Add a spmd for application to pass in topology (#5931)

#6054

Merged