PR #5129 Update Python device API for SPMD

Update Python device API for SPMD #5129

JackCaoG merged 2 commits into master from JackCao/spmd_python_device

JackCaoG added distributed

JackCaoG requested a review from

yeounoh 2 years ago

JackCaoG requested a review from

jonb377 2 years ago

JackCaoG requested a review from

will-cromar 2 years ago

alanwaketan requested a review from

alanwaketan 2 years ago

jonb377 commented on 2023-06-07

vanbasten23 commented on 2023-06-08

JackCaoG force pushed from a4bd5b61 to 596039f3 2 years ago

JackCaoG force pushed from cfe3ae75 to 616d7ce9 2 years ago

JackCaoG marked this pull request as ready for review 2 years ago

will-cromar commented on 2023-06-27

alanwaketan approved these changes on 2023-06-27

JackCaoG force pushed from 616d7ce9 to d31f11d7 2 years ago

JackCaoG force pushed from c304e067 to 6c0e8c92 2 years ago

jonb377 commented on 2023-07-12

Make python Api to respect the virtual device when SPMD is enabled

3b153b3e

JackCaoG force pushed from 6c0e8c92 to 3b153b3e 2 years ago

fix typo

a87db71a

JackCaoG requested a review from

will-cromar 2 years ago

JackCaoG requested a review from

jonb377 2 years ago

JackCaoG changed the title ~~Add Test for xla_model api + SPMD~~ Update Python device API for SPMD 2 years ago

jonb377 approved these changes on 2023-07-24

JackCaoG merged f996701a into master 2 years ago

Reviewers

jonb377

alanwaketan

will-cromar

vanbasten23

yeounoh

Assignees

No one assigned

Labels

distributed

Milestone

No milestone