xla
Implement collective gather op
#9435
Merged

Implement collective gather op #9435

bfolie merged 13 commits into master from bfolie/gather-collective
bfolie
bfolie first attempt at gather, hangs
5203f389
bfolie get gather working
e9fe828d
bfolie make gather work for coalesced inputs
45626874
bfolie add some more comments to test
8173e716
bfolie format
39f19084
bfolie move scatter and gather tests into more appropriate class
1a6c3b65
bfolie bfolie requested a review from pgmoka pgmoka 218 days ago
bfolie
bfolie commented on 2025-07-01
bfolie
bfolie commented on 2025-07-01
bfolie
bfolie bfolie requested a review from benawilson benawilson 217 days ago
pgmoka
pgmoka commented on 2025-07-08
bfolie Add space before comment
619d4435
bfolie bfolie enabled auto-merge (squash) 210 days ago
disabled auto-merge 210 days ago
Manually disabled by user
bfolie Merge branch 'master' of https://github.com/pytorch/xla into bfolie/g…
c57a5071
bfolie Merge branch 'bfolie/gather-collective' of https://github.com/pytorch…
e19d94fa
bfolie bfolie requested a review from pgmoka pgmoka 208 days ago
pgmoka
pgmoka approved these changes on 2025-07-14
bfolie Merge branch 'master' into bfolie/gather-collective
667bfb5a
bfolie Merge branch 'master' of https://github.com/pytorch/xla into bfolie/g…
5f51a14b
bfolie Merge branch 'bfolie/gather-collective' of https://github.com/pytorch…
cb355ab6
bfolie Update test_collective_ops_tpu.py
492e1ee6
bfolie bfolie merged a021bf09 into master 202 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone