xla
68d577ee - Add `runtime.host_index` (#5283)

Commit

2 years ago

Add `runtime.host_index` (#5283)

References

#5283 - Add `runtime.host_index`

Author

will-cromar

will-cromar

Parents

Files4

test/pjrt
- test_internal_tpu.py
- test_runtime.py
torch_xla
- _internal
  - tpu.py
- runtime.py