Aashiq Dheeraj bb6565ef14 add fftshift and ifftshift fft helpers (#2135) ...

* add fftshift and ifftshift fft helpers

* address comments

* axes have to be iterable

* fix fp error in roll + add test

---------

Co-authored-by: Aashiq Dheeraj <aashiq@aashiq-mbp-m4.local>

2025-04-29 22:13:45 -07: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

Dynamic slicing (#1741)

2025-01-07 14:02:16 -08:00

blas_tests.cpp

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

CMakeLists.txt

revise cmake minimum for doctest (#2014)

2025-03-27 19:30:58 -07:00

compile_tests.cpp

Simplify removes no-ops from the tape (#1759)

2025-01-09 11:23:19 -08:00

creations_tests.cpp

Remove test of converting negative float to uint (#2048)

2025-04-06 06:21:46 -07: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

export_import_tests.cpp

Use unordered map for kwargs in export/import (#2087)

2025-04-21 07:17:22 -07:00

fft_tests.cpp

add fftshift and ifftshift fft helpers (#2135)

2025-04-29 22:13:45 -07:00

linalg_tests.cpp

Adds nuclear norm support (#1894)

2025-03-04 13:26:02 -08:00

load_tests.cpp

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

metal_tests.cpp

move memory APIs into top level mlx.core (#1982)

2025-03-21 07:25:12 -07:00

ops_tests.cpp

add fftshift and ifftshift fft helpers (#2135)

2025-04-29 22:13:45 -07:00

random_tests.cpp

Resolved ambiguity in mlx::core::take_along_axis (#1822)

2025-02-04 06:06:17 -08:00

scheduler_tests.cpp

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

tests.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

utils_tests.cpp

iinfo and scalar overflow detection (#2009)

2025-03-27 19:54:56 -07:00

vmap_tests.cpp

Adds nuclear norm support (#1894)

2025-03-04 13:26:02 -08:00