George HotzandGitHub 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

..

__init__.py

update docs, remove corealize (#4264)

2024-04-23 12:05:29 +04:00

grouper.py

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

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

jit.py

jit: prune independent copies (#9749)

2025-04-05 20:50:28 +03:00

memory.py

jit: optimize before pickle (#9611)

2025-03-28 19:06:09 +07:00

multi.py

remove unneeded annotation import (#8934)

2025-02-06 13:12:35 -05:00

realize.py

bump assembly debug to 7 (#9662)

2025-04-01 11:51:33 +08:00

schedule.py

track_rewrites in scheduler [pr] (#9801)

2025-04-09 12:48:14 +08:00

search.py

pipe allow_test_size in _time_program (#9789)

2025-04-08 09:07:40 -04:00