mlx/python/src
Awni Hannun be98f4ab6b
Reduce a little overhead (#871)
* some small overhead improvements

* use result_type in rms_norm

* remove release force

* fix + use non-vector version

* revert compile change

* fix ops

* a little more overhead

* a little more cleanup and overhead
2024-03-22 17:29:36 -07:00
..
array.cpp Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
buffer.h Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
CMakeLists.txt Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
constants.cpp Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
convert.cpp Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
convert.h Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
device.cpp Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
fast.cpp Reduce a little overhead (#871) 2024-03-22 17:29:36 -07:00
fft.cpp Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
indexing.cpp Set item bug fix (#879) 2024-03-22 12:11:17 -07:00
indexing.h Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
linalg.cpp Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
load.cpp Do not store iostream in shared_ptr (#872) 2024-03-22 06:54:45 -07:00
load.h Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
metal.cpp Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
mlx.cpp Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
ops.cpp Bump (#869) 2024-03-21 13:56:56 -07:00
random.cpp Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
stream.cpp Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
transforms.cpp Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
trees.cpp Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
trees.h Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
utils.h Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00