zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-24 01:17:26 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update fft_tests.cpp

Browse Source
This commit is contained in:
Param Thakkar 2025-05-07 15:24:01 +05:30 committed by GitHub
parent 80002ed42f
commit eeaf1fa463
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/fft_tests.cpp
View File

@ -309,7 +309,6 @@ TEST_CASE("test fft grads") {
CHECK_EQ(vjp_out.shape(), Shape{5, 5});
}
<<<<<<< HEAD
TEST_CASE("test stft and istft") {
int n_fft = 4;
int hop_length = 2;
@ -383,7 +382,8 @@ TEST_CASE("test stft and istft") {
CHECK_EQ(stft_result.shape(1), n_fft);
}
}
== == == = TEST_CASE("test fftshift and ifftshift") {
TEST_CASE("test fftshift and ifftshift") {
// Test 1D array with even length
auto x = arange(8);
auto y = fft::fftshift(x);
@ -440,4 +440,3 @@ TEST_CASE("test stft and istft") {
CHECK_THROWS_AS(fft::ifftshift(x, {3}), std::invalid_argument);
CHECK_THROWS_AS(fft::ifftshift(x, {-5}), std::invalid_argument);
}
>>>>>>> 5a1a5d5ed16f69af7c3ce56dd94e4502661e1565