xla
Add initial support for `scatter_reduce`
#4517
Merged

Add initial support for `scatter_reduce` #4517

seanlatias
JackCaoG JackCaoG added lowering
JackCaoG
seanlatias
JackCaoG
seanlatias seanlatias force pushed from 511236dc to 5436a722 2 years ago
seanlatias seanlatias force pushed from 5436a722 to 6092b0af 2 years ago
JackCaoG JackCaoG requested a review from JackCaoG JackCaoG 2 years ago
JackCaoG
JackCaoG commented on 2023-01-31
JackCaoG
JackCaoG commented on 2023-01-31
JackCaoG
JackCaoG commented on 2023-01-31
JackCaoG
JackCaoG commented on 2023-01-31
JackCaoG
JackCaoG commented on 2023-01-31
initial support for scatter_reduce
4e8caa36
Added cpp tests
75144b8d
undo setup.py
780dd338
fix linter for cpp test
7e572c5a
move check_count out of for loops
17341d01
seanlatias seanlatias force pushed from d26a8f5a to 17341d01 2 years ago
JackCaoG
JackCaoG approved these changes on 2023-02-02
JackCaoG JackCaoG merged 99bdfe8d into master 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone