mlx/mlx
toji 85143fecdd
improved error msg for invalid axis(mx.split) (#685)
* improved error msg for invalid axis(`mx.split`)

* Apply suggestions from code review

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

* fixed formatting issue

---------

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
2024-02-15 07:25:38 -08:00
..
3rdparty jagrit's commit files 2023-11-29 10:52:08 -08:00
backend Custom primitive + RoPE fat op (#676) 2024-02-14 14:04:25 -08:00
io Metadata support for safetensors (#639) 2024-02-08 19:33:15 -08:00
types Remainder negative numerator bug fixed (#641) 2024-02-09 16:49:14 -08:00
allocator.cpp Memory allocation (#292) 2024-01-02 11:59:19 -08:00
allocator.h Memory allocation (#292) 2024-01-02 11:59:19 -08:00
array.cpp Fix empty array construction in cpp (#684) 2024-02-13 23:34:17 -08:00
array.h Fix empty array construction in cpp (#684) 2024-02-13 23:34:17 -08:00
CMakeLists.txt Custom primitive + RoPE fat op (#676) 2024-02-14 14:04:25 -08:00
compile.cpp Compile with capture (#629) 2024-02-07 17:29:22 -08:00
compile.h Compile primitive (#571) 2024-02-05 06:51:22 -08:00
device.cpp copyright + ack 2023-11-30 11:12:53 -08:00
device.h copyright + ack 2023-11-30 11:12:53 -08:00
dtype.cpp copyright + ack 2023-11-30 11:12:53 -08:00
dtype.h random generation fix (#80) 2023-12-08 10:40:57 -08:00
fast.cpp Custom primitive + RoPE fat op (#676) 2024-02-14 14:04:25 -08:00
fast.h Custom primitive + RoPE fat op (#676) 2024-02-14 14:04:25 -08:00
fft.cpp Spelling (#342) 2024-01-01 21:08:17 -08:00
fft.h copyright + ack 2023-11-30 11:12:53 -08:00
graph_utils.cpp Kernel generation (#614) 2024-02-07 13:15:59 -08:00
graph_utils.h Kernel generation (#614) 2024-02-07 13:15:59 -08:00
io.h Metadata support for safetensors (#639) 2024-02-08 19:33:15 -08:00
linalg.cpp QR factorization (#310) 2024-01-26 09:27:31 -08:00
linalg.h QR factorization (#310) 2024-01-26 09:27:31 -08:00
mlx.h Custom primitive + RoPE fat op (#676) 2024-02-14 14:04:25 -08:00
ops.cpp improved error msg for invalid axis(mx.split) (#685) 2024-02-15 07:25:38 -08:00
ops.h Adds device context manager (#679) 2024-02-14 14:14:58 -08:00
primitives.cpp Reduce vmap + some fixes (#601) 2024-02-01 11:30:28 -08:00
primitives.h Kernel generation (#614) 2024-02-07 13:15:59 -08:00
random.cpp Add loc and scale to random.normal (#638) 2024-02-07 11:49:59 -08:00
random.h Add loc and scale to random.normal (#638) 2024-02-07 11:49:59 -08:00
scheduler.cpp copyright + ack 2023-11-30 11:12:53 -08:00
scheduler.h move all ObjC (via metal-cpp) interaction until post static initializers (#370) 2024-01-04 16:12:00 -08:00
stream.h copyright + ack 2023-11-30 11:12:53 -08:00
transforms_impl.h Compile front-end (#476) 2024-01-26 13:45:30 -08:00
transforms.cpp Reduce vmap + some fixes (#601) 2024-02-01 11:30:28 -08:00
transforms.h Compile primitive (#571) 2024-02-05 06:51:22 -08:00
utils.cpp Adds device context manager (#679) 2024-02-14 14:14:58 -08:00
utils.h Adds device context manager (#679) 2024-02-14 14:14:58 -08:00