PR #4038 Use PjRt GPU client

Use PjRt GPU client #4038

jonb377 merged 2 commits into master from jonbolin-gpu-client

jonb377 requested a review from

will-cromar 3 years ago

jonb377 requested a review from

darisoy 3 years ago

JackCaoG added runtime

jonb377 force pushed from f1951e48 to f4c47f98 3 years ago

will-cromar requested changes on 2022-09-23

jonb377 force pushed from 24a937ce to c8d53bc0 3 years ago

will-cromar force pushed from c8d53bc0 to 8a5568bf 3 years ago

jonb377 force pushed from 8a5568bf to ae0ac198 3 years ago

Use PjRt GPU client

cce6bc01

Only allow access to GPU device 0

ddd1e2ba

jonb377 force pushed from ae0ac198 to ddd1e2ba 3 years ago

will-cromar approved these changes on 2022-10-14

jonb377 merged c58480a0 into master 3 years ago

jonb377 deleted the jonbolin-gpu-client branch 3 years ago

JackCaoG approved these changes on 2022-10-14

Reviewers

JackCaoG

will-cromar

darisoy

Assignees

No one assigned

Labels

runtime

Milestone

No milestone