xla
[PJRT] Cache world size and ordinal only when addressable_device_count == 1
#4965

Merged

[PJRT] Cache world size and ordinal only when addressable_device_count == 1 #4965

will-cromar merged 1 commit into master from wcromar/check-device-count

[PJRT] Check addressable device count instead of TPU version

f8b17bb4

will-cromar requested a review from

alanwaketan 2 years ago

will-cromar requested a review from

JackCaoG 2 years ago

alanwaketan approved these changes on 2023-05-02

JackCaoG approved these changes on 2023-05-02

will-cromar merged 5aa1ac3d into master 2 years ago

Reviewers

JackCaoG

alanwaketan

Assignees

No one assigned

Labels

None yet

Milestone

No milestone