mlx/tests
Awni Hannun 42afe27e12
std and expm1 (#973)
* std and expm1

* actually add expm1

* fix linux

* fix vjp

* relax tol for linux test

* Add it to the compilable primitives

---------

Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
2024-04-08 14:26:01 -07:00
..
allocator_tests.cpp Metal validation (#432) 2024-01-11 11:57:24 -08:00
arg_reduce_tests.cpp Don't use make_unique to create shared_ptr (#902) 2024-03-27 06:13:29 -07:00
array_tests.cpp Fix empty array construction in cpp (#684) 2024-02-13 23:34:17 -08:00
autograd_tests.cpp Implement the 'where' primitive for conditional selection (#664) 2024-02-22 15:10:48 -08:00
blas_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
CMakeLists.txt discover doctests in cmake (#703) 2024-02-19 07:03:56 -08:00
compile_tests.cpp Fixes #918 bug in compile_tests (#919) 2024-03-27 22:37:37 -07: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 Add matrix inversion primitive (#822) 2024-03-15 06:34:36 -07:00
load_tests.cpp Metadata support for safetensors (#639) 2024-02-08 19:33:15 -08:00
metal_tests.cpp bindings for memory info (#761) 2024-03-01 19:51:58 -08:00
ops_tests.cpp std and expm1 (#973) 2024-04-08 14:26:01 -07:00
random_tests.cpp Pickle + dtype fix for numpy conversion (#763) 2024-03-02 06:09:29 -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 Ensure shape dimensions are within supported integer range (#566) (#704) 2024-03-25 13:29:45 -07:00
vmap_tests.cpp Add vmap for SVD and inverse (#849) 2024-03-21 13:18:27 -07:00