mlx/tests
Awni Hannun 7c441600fe
Compile stride bug (#812)
* fix compile stride bug

* revert sdpa fix

* fix cpu

* fix bug with simplifying outputs
2024-03-11 06:31:31 -07: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 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 Compile stride bug (#812) 2024-03-11 06:31:31 -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 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 bindings for memory info (#761) 2024-03-01 19:51:58 -08:00
ops_tests.cpp nice tensordot for mlx c (#782) 2024-03-04 09:51:02 -08: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 Added mx.stack c++ frontend impl (#123) 2023-12-14 13:21:19 -08:00
vmap_tests.cpp Implement the 'where' primitive for conditional selection (#664) 2024-02-22 15:10:48 -08:00