mlx/tests
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
..
allocator_tests.cpp Metal validation (#432) 2024-01-11 11:57:24 -08:00
arg_reduce_tests.cpp Metal validation (#432) 2024-01-11 11:57:24 -08:00
array_tests.cpp Fix empty array construction in cpp (#684) 2024-02-13 23:34:17 -08:00
autograd_tests.cpp Split multi output (#461) 2024-01-16 13:33:55 -08:00
blas_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
CMakeLists.txt Custom VJP and checkpointing (#541) 2024-01-30 16:04:45 -08:00
compile_tests.cpp CPU compile (#691) 2024-02-17 06:54:32 -08:00
creations_tests.cpp Propagate nans in binary ops (#579) 2024-01-29 11:19:38 -08:00
custom_vjp_tests.cpp Custom VJP and checkpointing (#541) 2024-01-30 16:04:45 -08:00
device_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
eval_tests.cpp Removes the retain_graph flag (#385) 2024-01-07 15:16:51 -08:00
fft_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
linalg_tests.cpp QR factorization (#310) 2024-01-26 09:27:31 -08:00
load_tests.cpp Metadata support for safetensors (#639) 2024-02-08 19:33:15 -08:00
metal_tests.cpp GGUF: Avoid dequantization when format is compatible (#426) 2024-01-23 15:43:57 -08:00
ops_tests.cpp Implement diagonal operator (#562) 2024-01-30 09:45:48 -08:00
random_tests.cpp random.uniform must respect dtype, even if lower precision than "low" (#280) 2023-12-24 07:04:43 -08:00
scheduler_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
utils_tests.cpp Added mx.stack c++ frontend impl (#123) 2023-12-14 13:21:19 -08:00
vmap_tests.cpp Removes the retain_graph flag (#385) 2024-01-07 15:16:51 -08:00