mlx/mlx/backend
Awni Hannun 5798256fcf
Shapeless compilation for some graphs (#687)
* shapeless compilation for some graphs

* update compile benchmark

* default compile a few activations

* buffer donation

* bugfix

* shapeless fix

* update tests to work for cpu and gpu fusion

* test kwargs

* add kwargs to compile

* Recompile when python arguments change

* no compile for tanh

* some constant tests

---------

Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
2024-02-19 21:43:54 -08:00
..
accelerate Refactor CPU compile preamble (#708) 2024-02-19 06:12:53 -08:00
common Shapeless compilation for some graphs (#687) 2024-02-19 21:43:54 -08:00
metal Shapeless compilation for some graphs (#687) 2024-02-19 21:43:54 -08:00
no_metal Separate fast ops and primitives (#699) 2024-02-16 19:16:39 -08:00