xla
1e0bb990 - Set PJRT device assignment in order of device ID (#3902)

Commit

3 years ago

Set PJRT device assignment in order of device ID (#3902) * Set device assignment in order of device ID. * Set thread default device in `run_multiprocess`

References

#3902 - Set PJRT device assignment in order of device ID

Author

will-cromar

Parents

f82e5d8a

xla 1e0bb990 - Set PJRT device assignment in order of device ID (#3902)

xla
1e0bb990 - Set PJRT device assignment in order of device ID (#3902)