KFR: skip real fft for odd sizes

kfrlib · Feb 26, 2024 · 779c730 · 779c730
1 parent bbcf60d
commit 779c730
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/fftlibs/kfr/kfr.cpp b/fftlibs/kfr/kfr.cpp
@@ -130,6 +130,12 @@ class fft_implementation<dims, real, false, invert, inplace> : public fft_impl<r
 template <typename real>
 fft_impl_ptr<real> fft_create(const std::vector<size_t>& size, bool is_complex, bool invert, bool inplace)
 {
+    if (!is_complex)
+    {
+        size_t s = std::accumulate(size.begin(), size.end(), size_t(1), std::multiplies<>{});
+        if (s & 1)
+            return nullptr;
+    }
     return fft_create_for<fft_implementation, real>(size, is_complex, invert, inplace);
 }
 

diff --git a/src/fft_benchmark.cpp b/src/fft_benchmark.cpp
@@ -154,7 +154,7 @@ std::string execfile(std::string command)
 static std::string outname;
 static bool progress = true;
 static bool banner   = true;
-bool avx2only = false;
+bool avx2only        = false;
 static std::vector<std::vector<size_t>> sizes;
 static std::vector<bool> is_complex_list{ true, false };
 static std::vector<bool> inverse_list{ false, true };