Commit 7abdeeb

authored and

Google-ML-Automation

committed

Adding gather reduce sparse core kernel.

PiperOrigin-RevId: 889292646

1 parent 5d16abb commit 7abdeebCopy full SHA for 7abdeeb

2 files changed

+1840

-0

lines changed

src/maxtext/kernels
- gather_reduce_sc.py
tests
- gather_reduce_sc_test.py

2 files changed

+1840

-0

lines changed

Comments

(0)