zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-01 12:49:44 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add fftshift and ifftshift fft helpers (#2135)

Browse Source 
* 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>
This commit is contained in:
Aashiq Dheeraj
2025-04-30 01:13:45 -04:00
committed by GitHub
parent 7bb063bcb3
commit bb6565ef14
9 changed files with 275 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/ops_tests.cpp
View File

@@ -3859,6 +3859,9 @@ TEST_CASE("test roll") {
y = roll(x, {1, 2}, {0, 1});
CHECK(array_equal(y, array({8, 9, 5, 6, 7, 3, 4, 0, 1, 2}, {2, 5}))
.item<bool>());
y = roll(array({}), 0, 0);
CHECK(array_equal(y, array({})).item<bool>());
}
TEST_CASE("test contiguous") {