mlx/python
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
..
mlx Shapeless compilation for some graphs (#687) 2024-02-19 21:43:54 -08:00
src Shapeless compilation for some graphs (#687) 2024-02-19 21:43:54 -08:00
tests Shapeless compilation for some graphs (#687) 2024-02-19 21:43:54 -08:00