xla
Correctly configure PJRT execution when nprocs=1.
#4285

Merged

Correctly configure PJRT execution when nprocs=1. #4285

stgpetrovic merged 7 commits into master from pjrt-single-proc

JackCaoG requested a review from

will-cromar 3 years ago

JackCaoG added runtime

will-cromar requested changes on 2022-12-06

stgpetrovic force pushed from 3f4bc40c to 57d8f260 3 years ago

stgpetrovic requested a review from

will-cromar 3 years ago

stgpetrovic force pushed from 0aca236d to e49fbdc2 3 years ago

will-cromar approved these changes on 2022-12-12

Correctly configure PJRT execution when nprocs=1.

4eccd843

Format python

351debfa

Incorporate review feedback

0f2a88bd

Use executor until more review comments come in

fad534b8

Update pjrt world size test.

d9c976db

Remove subprocess executors for nprocs=1.

db579f8b

Cleanup duplicate code and unneeded wrapper.

d8cb6fcb

stgpetrovic force pushed from 855444e5 to d8cb6fcb 3 years ago

stgpetrovic merged 35a5dc12 into master 3 years ago

stgpetrovic deleted the pjrt-single-proc branch 3 years ago

Reviewers

will-cromar

JackCaoG

Assignees

No one assigned

Labels

runtime

Milestone

No milestone