xla
[PJRT] Flatten output of `pjrt.run_multiprocess`
#3996

Merged

[PJRT] Flatten output of `pjrt.run_multiprocess` #3996

will-cromar merged 11 commits into master from pjrt-flatten-multiprocess

will-cromar changed the title ~~Flatten output of `pjrt.run_multiprocess`~~ [PJRT] Flatten output of `pjrt.run_multiprocess` 3 years ago

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

will-cromar requested a review from

yeounoh 3 years ago

will-cromar requested a review from

JackCaoG 3 years ago

will-cromar requested a review from

alanwaketan 3 years ago

will-cromar force pushed from 0c1847c3 to 02efc3b3 3 years ago

JackCaoG commented on 2022-09-14

Flatten output of `pjrt.run_multiprocess`.

3f5c035e

Update tests

2942f635

spawn support

681c46e3

Call from `xla_multiprocess`

bf3cda07

Spawn test

8db5d0c2

Formatting

d04dce83

make signature compatible with python 3.7

cf198c8a

Clean up getting device ordinal

f2ce3218

will-cromar force pushed from 02efc3b3 to f2ce3218 3 years ago

JackCaoG approved these changes on 2022-09-15

formatting

1724d9c3

fix global flags variable

f1318bfb

formatting

3cd462c1

will-cromar merged eecf89a2 into master 3 years ago

Reviewers

JackCaoG

yeounoh

alanwaketan

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

xla [PJRT] Flatten output of `pjrt.run_multiprocess` #3996 Merged

[PJRT] Flatten output of `pjrt.run_multiprocess` #3996

xla
[PJRT] Flatten output of `pjrt.run_multiprocess`
#3996

Merged