xla
Implement `xm.rendezvous` with XLA collective communication
#4181
Merged

Implement `xm.rendezvous` with XLA collective communication #4181

will-cromar merged 6 commits into master from wcromar/xla-rendezvous
will-cromar
ronghanghu
ronghanghu commented on 2022-11-10
will-cromar Implement pjrt.rendezvous with XLA collective ops.
3c1555c9
will-cromar Update tests.
9fe60135
will-cromar Formatting
bd2ca0c2
will-cromar Handle some edge cases better
f66a6e87
will-cromar formatting
585982df
will-cromar will-cromar force pushed from 78ecc43f to 585982df 3 years ago
will-cromar will-cromar changed the title [WIP] Implement `xm.rendezvous` with XLA collective communication Implement `xm.rendezvous` with XLA collective communication 3 years ago
will-cromar will-cromar requested a review from JackCaoG JackCaoG 3 years ago
will-cromar will-cromar marked this pull request as ready for review 3 years ago
AlexWertheim
JackCaoG
JackCaoG commented on 2022-11-11
will-cromar Add mesh service test to CI
91207682
will-cromar will-cromar requested a review from JackCaoG JackCaoG 3 years ago
JackCaoG
JackCaoG approved these changes on 2022-11-15
will-cromar
will-cromar will-cromar added runtime
will-cromar will-cromar merged a9eb3459 into master 3 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone