zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-22 01:21:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
472 Commits 37 Branches 67 Tags 818 MiB
ae18326533
Commit Graph

2 Commits

Author SHA1 Message Date
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
Awni Hannun
d75ae52ecd
Compile primitive (#571) 
* Compiled primitive with basic binary, unary graph-level fusion
 2024-02-05 06:51:22 -08:00