George Hotz fce432d2e3 Ops.FUSE makes softmax a single kernel (#9808) ...

* KERNELIZE makes softmax a single kernel

* single kernel works

* softmax works

* broken

* correct

* skip that test

* kernelize tests

* rename to fuse

* better reduce_push_add_ones code

* correct now

* cleanups

* oops

* return None if we can't push ones

* rename + docs

* atol fixes group

* flash attention broken test

2025-04-09 22:56:28 +08:00

3.6 KiB

Raw Blame History

View Raw