xla
[PJRT] Defer default device selection until we need to know the runtime
#4630
Merged

[PJRT] Defer default device selection until we need to know the runtime #4630

will-cromar merged 9 commits into master from wcromar/fix-default-pjrt
will-cromar
will-cromar Defer default device selection until we need to know the runtime
52dff3e9
will-cromar will-cromar added runtime
will-cromar formatting
f5be92ea
will-cromar More formatting
dc984c02
will-cromar
will-cromar Reimplement in python
9523c55e
will-cromar Revert C++ changes
cd6b5159
will-cromar Use xenv variable
e2643c7a
will-cromar Add to configuration.yaml
14d27902
will-cromar will-cromar requested a review from JackCaoG JackCaoG 3 years ago
will-cromar will-cromar marked this pull request as ready for review 3 years ago
JackCaoG
JackCaoG commented on 2023-02-16
JackCaoG
JackCaoG approved these changes on 2023-02-16
will-cromar add xla_dist test case
62c5ca61
will-cromar formatting
9554fe45
will-cromar
will-cromar will-cromar merged 32da64f6 into master 3 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone