xla
7f4d1903 - Move PJRT Python APIs out of `torch_xla.experimental.*` (#5011)

Commit
2 years ago
Move PJRT Python APIs out of `torch_xla.experimental.*` (#5011) * Move torch_xla.experimental.pjrt * Move deprecation to common module * Move tpu.py * Move gpu.py * pjrt -> this_module * formatting * formatting better * Move run_multiprocess * Start migrating tests * Add an __init__.py file * formatting * Fix import * Replace more instances of torch_xla.experimental * Fix import in pjrt_backend * Fix import in test_ddp * Move pjrt rendezvous * Move CI tests * formatting * Fix TPU integration test name * Adding missing multi_cpu test * fix test name typo * Merge global_device_attributes * Update PJRT warning * xr.requires_pjrt * xr.global_device_count
Author
Parents
Loading