xla
[PJRT] Update `torch.distributed` unit test to use PJRT
#4743
Merged
Overview
Commits
9
Changes
View On GitHub

[PJRT] Update `torch.distributed` unit test to use PJRT #4743

will-cromar merged 9 commits into master from wcromar/update-dist-test
will-cromar
will-cromar Start refactoring torch.distributed test to support PJRT
69426937
will-cromar Fix imports
0d79acf0
will-cromar Use unit test assertions
926f1d17
will-cromar pjrt init method
8efc331c
will-cromar Add back class method
809d12a5
will-cromar Formatting
a7bec44e
will-cromar will-cromar added runtime
will-cromar
will-cromar Fix Send and Recv tests
3da73b9f
will-cromar Remove unused import
1710ecba
will-cromar Use dist.send/recv
6ff5ffa0
will-cromar will-cromar changed the title [WIP] Update `torch.distributed` unit test to use PJRT [PJRT] Update `torch.distributed` unit test to use PJRT 2 years ago
will-cromar will-cromar marked this pull request as ready for review 2 years ago
will-cromar will-cromar requested a review from JackCaoG JackCaoG 2 years ago
will-cromar will-cromar requested a review from alanwaketan alanwaketan 2 years ago
alanwaketan
alanwaketan approved these changes on 2023-03-09
will-cromar will-cromar merged 5da15023 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