xla
Set PJRT device assignment in order of device ID
#3902

Merged

Set PJRT device assignment in order of device ID #3902

will-cromar merged 5 commits into master from pjrt-collective-rank

will-cromar marked this pull request as ready for review 3 years ago

will-cromar requested a review from

JackCaoG 3 years ago

will-cromar requested a review from

darisoy 3 years ago

will-cromar requested a review from

yeounoh 3 years ago

JackCaoG commented on 2022-08-19

JackCaoG approved these changes on 2022-08-22

Set device assignment in order of device ID.

9a6a9dc5

Set thread default device in `run_multiprocess`, not `xla_device`

c596f498

Add collective operation tests for PjRt

d443f34b

Formatting

2648029a

Add default device back to `pjrt.xla_device`

887c00cf

will-cromar force pushed from cc0c387c to 887c00cf 3 years ago

will-cromar added runtime

will-cromar merged 1e0bb990 into master 3 years ago

Reviewers

JackCaoG

darisoy

yeounoh

Assignees

No one assigned

Labels

runtime

Milestone

No milestone