Skip to content

Implement collective gather op#9435

Merged
bfolie merged 13 commits intomasterfrom
bfolie/gather-collective
Jul 18, 2025
Merged

Implement collective gather op#9435
bfolie merged 13 commits intomasterfrom
bfolie/gather-collective

Conversation

@bfolie
Copy link
Copy Markdown
Collaborator

@bfolie bfolie commented Jul 1, 2025

@bfolie bfolie requested a review from pgmoka July 1, 2025 23:08
Comment thread test/pjrt/test_collective_ops_tpu.py Outdated
Comment thread test/pjrt/test_collective_ops_tpu.py
@bfolie
Copy link
Copy Markdown
Collaborator Author

bfolie commented Jul 2, 2025

Failing tests are expected until the TPU CI cluster is updated to use python 3.12. See #9434

@bfolie bfolie requested a review from benawilson July 2, 2025 19:31
Comment thread torch_xla/distributed/xla_backend.py
Comment thread torch_xla/distributed/xla_backend.py
Comment thread torch_xla/distributed/xla_backend.py
@bfolie bfolie enabled auto-merge (squash) July 9, 2025 20:06
@bfolie bfolie disabled auto-merge July 9, 2025 20:06
@bfolie bfolie requested a review from pgmoka July 11, 2025 18:13
Comment thread torch_xla/distributed/xla_backend.py
Comment thread torch_xla/distributed/xla_backend.py
@bfolie bfolie merged commit a021bf0 into master Jul 18, 2025
23 of 24 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants