mirror of
https://github.com/ml-explore/mlx.git
synced 2025-07-19 07:31:26 +08:00
5798256fcf
* 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> |
||
|---|---|---|
| .. | ||
| array.cpp | ||
| CMakeLists.txt | ||
| constants.cpp | ||
| device.cpp | ||
| fast.cpp | ||
| fft.cpp | ||
| indexing.cpp | ||
| indexing.h | ||
| linalg.cpp | ||
| load.cpp | ||
| load.h | ||
| metal.cpp | ||
| mlx.cpp | ||
| ops.cpp | ||
| random.cpp | ||
| stream.cpp | ||
| transforms.cpp | ||
| utils.cpp | ||
| utils.h | ||