mlx/mlx/backend/common
Awni Hannun dc937b8ed3
CPU compile (#691)
* build and load shared object for cpu compile

* nits

* cpu compile tests pass

* cpu compile tests pass

* fix preamble for g++

* donation

* fix gpu buffer donation

* reuse prebuilt libraries

* faster contiguity conditoins

* fix test

* rid compiler warning

* fast erf

* Fix float16 for compile and add more types to cpu compile

* Remove a forgotten comment

* use cached libs

* nits

---------

Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
2024-02-17 06:54:32 -08:00
..
arange.h copyright + ack 2023-11-30 11:12:53 -08:00
arg_reduce.cpp copyright + ack 2023-11-30 11:12:53 -08:00
binary_two.h Multi output primitives (#330) 2024-01-08 16:39:08 -08:00
binary.cpp Remainder negative numerator bug fixed (#641) 2024-02-09 16:49:14 -08:00
binary.h Buffer Donation (#519) 2024-01-26 16:30:33 -08:00
CMakeLists.txt Custom primitive + RoPE fat op (#676) 2024-02-14 14:04:25 -08:00
compiled_preamble.h CPU compile (#691) 2024-02-17 06:54:32 -08:00
compiled.cpp CPU compile (#691) 2024-02-17 06:54:32 -08:00
compiled.h CPU compile (#691) 2024-02-17 06:54:32 -08:00
conv.cpp fixes for building and running on iOS (#619) 2024-02-04 12:29:17 -08:00
copy.cpp Buffer Donation (#519) 2024-01-26 16:30:33 -08:00
copy.h copyright + ack 2023-11-30 11:12:53 -08:00
default_primitives.cpp CPU compile (#691) 2024-02-17 06:54:32 -08:00
erf.cpp copyright + ack 2023-11-30 11:12:53 -08:00
erf.h format 2023-11-30 11:50:50 -08:00
fft.cpp copyright + ack 2023-11-30 11:12:53 -08:00
indexing.cpp copyright + ack 2023-11-30 11:12:53 -08:00
load.cpp Spelling (#342) 2024-01-01 21:08:17 -08:00
primitives.cpp Custom VJP and checkpointing (#541) 2024-01-30 16:04:45 -08:00
qrf.cpp fixes for building and running on iOS (#619) 2024-02-04 12:29:17 -08:00
quantized.cpp Custom VJP and checkpointing (#541) 2024-01-30 16:04:45 -08:00
reduce.cpp copyright + ack 2023-11-30 11:12:53 -08:00
reduce.h Refactor the reduction kernels (#277) 2023-12-24 14:47:57 -08:00
rope.cpp Separate fast ops and primitives (#699) 2024-02-16 19:16:39 -08:00
scan.cpp copyright + ack 2023-11-30 11:12:53 -08:00
softmax.cpp Fix the softmax fix (#661) 2024-02-09 17:02:13 -08:00
sort.cpp copyright + ack 2023-11-30 11:12:53 -08:00
threefry.cpp copyright + ack 2023-11-30 11:12:53 -08:00
threefry.h copyright + ack 2023-11-30 11:12:53 -08:00
unary.h Buffer Donation (#519) 2024-01-26 16:30:33 -08:00
utils.h copyright + ack 2023-11-30 11:12:53 -08:00