Awni Hannun 2419edd5b2 Faster indexing math in a few kernels (#1589) ...

* wip: faster compiled kernels

* faster general unary with uint specialization

* index type in compiled, unary, binary, ternary, copy

* fix jit

* jit fix

* specialize gather + scatter

* nit in docs

2024-11-18 19:52:00 -08:00

5.7 KiB

Raw Blame History

View Raw