xla

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

[Backport]Add a spmd for application to pass in topology (#5931) (#6054) Co-authored-by: amithrm <84203728+amithrm@users.noreply.github.com> Co-authored-by: Shashwat Srijan <119712013+sssrijan-amazon@users.noreply.github.com>