xla
f6ff59f8
- Cache world_size and ordinal in pjrt
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
Cache world_size and ordinal in pjrt
References
#4922 - [Distributed] Make xm.all_gather a single graph in Dynamo
Author
alanwaketan
Committer
alanwaketan
Parents
a432f999
Loading
