search

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.49k stars 480 forks source link

cherry pick collective op all_to_all_single #8077

Closed zpcore closed 1 month ago

zpcore commented 1 month ago

There is a lint issue in the torch_xla/__init__.py file. Updated the file format in the PR also.