Files
mlx/mlx/backend/metal
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
..
2024-02-17 06:54:32 -08:00
2023-11-30 11:12:53 -08:00
2024-01-02 11:59:19 -08:00
2024-02-07 13:15:59 -08:00
2024-02-18 12:50:10 -08:00
2024-01-26 16:30:33 -08:00
2023-11-30 11:12:53 -08:00
2024-02-07 13:15:59 -08:00
2023-11-30 11:12:53 -08:00
2024-02-18 12:50:10 -08:00
2024-01-30 16:04:45 -08:00
2024-01-17 12:42:39 -08:00
2024-02-02 09:57:12 -08:00
2024-02-18 12:50:10 -08:00
2024-02-12 18:54:21 -08:00
2023-11-30 11:12:53 -08:00
2024-02-09 17:02:13 -08:00
2024-01-01 21:08:17 -08:00