mlx/tests
Awni Hannun 4e1e9520e1
Flatten and unflatten (#1692)
* flatten and unflatten

* fix grad

* fix shape infer

* use squeeze + unsqueeze in get_item
2024-12-11 21:51:37 -08:00
..
allocator_tests.cpp Metal validation (#432) 2024-01-11 11:57:24 -08:00
arg_reduce_tests.cpp Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00
array_tests.cpp Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00
autograd_tests.cpp Fix x86 tests (#1691) 2024-12-11 07:47:18 -08:00
blas_tests.cpp Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00
CMakeLists.txt Chore: add pre-commit hook for cmake (#1362) 2024-09-16 12:53:01 -07:00
compile_tests.cpp Using math defines requires switch in MSVC (#1665) 2024-12-08 08:16:28 -08:00
creations_tests.cpp Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00
custom_vjp_tests.cpp Update pre-commit hooks (#984) 2024-04-11 07:27:53 -07:00
device_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
einsum_tests.cpp Einsum (#1269) 2024-07-25 09:36:44 -07:00
eval_tests.cpp Shared events for synchronization + async eval (#998) 2024-04-17 06:16:02 -07:00
fft_tests.cpp Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00
linalg_tests.cpp Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00
load_tests.cpp Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00
metal_tests.cpp Expose function to clear memory cache (#1032) 2024-04-24 16:48:51 -07:00
ops_tests.cpp Flatten and unflatten (#1692) 2024-12-11 21:51:37 -08:00
random_tests.cpp Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -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 Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00
vmap_tests.cpp Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00