xla
[PJRT] Cherry pick local rank changes to r2.0 release
#4712
Merged

[PJRT] Cherry pick local rank changes to r2.0 release #4712

will-cromar merged 3 commits into r2.0 from wcromar/r2.0-local-rank
will-cromar
will-cromar [PJRT] Change the name of `LOCAL_RANK` to fix conflicts (#4646)
8b8ac8ff
will-cromar [PJRT] Use the new local rank env var to calculate local_ordinal (#4684)
d8c0e2ee
will-cromar [PJRT] Remove start method from _run_singleprocess (#4709)
46b249e2
will-cromar will-cromar requested a review from JackCaoG JackCaoG 2 years ago
will-cromar will-cromar added runtime
JackCaoG
JackCaoG approved these changes on 2023-03-02
will-cromar will-cromar merged 8af4ca19 into r2.0 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone