xla
Move PJRT Python APIs out of `torch_xla.experimental.*`
#5011
Merged
Overview
Commits
25
Changes
View On GitHub

Move PJRT Python APIs out of `torch_xla.experimental.*` #5011

will-cromar merged 25 commits into master from wcromar/stable-pjrt-api
will-cromar
will-cromar will-cromar changed the title [WIP] Move PJRT Python APIs out of `torch_xla.experimental.*` Move PJRT Python APIs out of `torch_xla.experimental.*` 2 years ago
will-cromar will-cromar requested a review from JackCaoG JackCaoG 2 years ago
will-cromar will-cromar marked this pull request as ready for review 2 years ago
JackCaoG
JackCaoG commented on 2023-05-19
JackCaoG
JackCaoG commented on 2023-05-19
will-cromar will-cromar force pushed from 8fa857fe to 660a4cf2 2 years ago
will-cromar will-cromar force pushed from 81046273 to 93def626 2 years ago
will-cromar will-cromar force pushed from 93def626 to de2f1174 2 years ago
will-cromar will-cromar requested a review from JackCaoG JackCaoG 2 years ago
JackCaoG JackCaoG added runtime
JackCaoG
JackCaoG commented on 2023-06-02
JackCaoG
JackCaoG approved these changes on 2023-06-02
will-cromar Move torch_xla.experimental.pjrt
4f5008b9
will-cromar Move deprecation to common module
e9fa31cd
will-cromar Move tpu.py
462f02a2
will-cromar Move gpu.py
e5637d9a
will-cromar pjrt -> this_module
6d7b4c2c
will-cromar formatting
5361fcb3
will-cromar formatting better
2bb37652
will-cromar Move run_multiprocess
abcd3973
will-cromar Start migrating tests
b27a50ba
will-cromar Add an __init__.py file
bd210513
will-cromar formatting
248003bf
will-cromar Fix import
21705a12
will-cromar Replace more instances of torch_xla.experimental
414c4e30
will-cromar Fix import in pjrt_backend
d0aee7f3
will-cromar Fix import in test_ddp
d25744a1
will-cromar Move pjrt rendezvous
e4e104a2
will-cromar Move CI tests
bfb9d266
will-cromar formatting
fb69f7d4
will-cromar Fix TPU integration test name
79bcf7ba
will-cromar Adding missing multi_cpu test
f5c8024c
will-cromar fix test name typo
c82883ad
will-cromar Merge global_device_attributes
8db53769
will-cromar Update PJRT warning
5e8fea71
will-cromar xr.requires_pjrt
f11be837
will-cromar will-cromar force pushed from dc1130e8 to f11be837 2 years ago
will-cromar xr.global_device_count
38d068b6
will-cromar will-cromar merged 7f4d1903 into master 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
runtime
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum