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

Merged

Commits

[PJRT] Change the name of `LOCAL_RANK` to fix conflicts (#4646)

will-cromar committed 3 years ago
[PJRT] Use the new local rank env var to calculate local_ordinal (#4684)

will-cromar committed 3 years ago
[PJRT] Remove start method from _run_singleprocess (#4709)

will-cromar committed 3 years ago

Loading