xla
Support multiple GPUs on a single host with PjRt
#4297

Merged

Support multiple GPUs on a single host with PjRt #4297

jonb377 merged 1 commit into master from jonbolin-pjrt-gpu

jonb377 requested a review from

will-cromar 3 years ago

jonb377 requested a review from

JackCaoG 3 years ago

JackCaoG added xla:gpu

JackCaoG added runtime

jonb377 force pushed from 7fc306bc to 49ab8bd8 3 years ago

jonb377 closed this 3 years ago

jonb377 reopened this 3 years ago

jonb377 marked this pull request as draft 3 years ago

jonb377 force pushed from 49ab8bd8 to 95d57cff 3 years ago

jonb377 force pushed from ff5b7788 to 27e5175d 2 years ago

jonb377 force pushed from 27e5175d to 3036d358 2 years ago

jonb377 force pushed from 3036d358 to f35f1260 2 years ago

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

will-cromar approved these changes on 2023-02-08

will-cromar requested a review from

steventk-g 2 years ago

steventk-g commented on 2023-02-08

steventk-g approved these changes on 2023-02-08

jonb377 force pushed from f35f1260 to 66fb5799 2 years ago

Support multiple GPUs on a single host with PjRt

4d2f200e

jonb377 force pushed from 66fb5799 to 4d2f200e 2 years ago

jonb377 merged 1e011c60 into master 2 years ago

jonb377 deleted the jonbolin-pjrt-gpu branch 2 years ago

Reviewers

steventk-g

will-cromar

JackCaoG

Assignees

No one assigned

Labels

xla:gpu runtime

Milestone

No milestone