MNN:Bugfix: Fix compile bug for windows

alibaba · Dec 31, 2024 · 42b37ac · 42b37ac
1 parent 27da5e7
commit 42b37ac
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/source/backend/cpu/CPUStft.cpp b/source/backend/cpu/CPUStft.cpp
@@ -11,6 +11,7 @@
 #define M_PI 3.141592654
 #endif
 #include <algorithm>
+#include <cmath>
 #include "backend/cpu/CPUStft.hpp"
 #include "backend/cpu/CPUBackend.hpp"
 #include "core/Concurrency.h"
@@ -27,10 +28,10 @@ static void MNNDftAbs(const float* input, const float* window, float* output, fl
         float real_sum = 0.f, imag_sum = 0.f;
         for (int n = 0; n < nfft; ++n) {
             float angle = 2 * M_PI * k * n / nfft;
-            real_sum += buffer[n] * std::cos(angle);
-            imag_sum -= buffer[n] * std::sin(angle);
+            real_sum += buffer[n] * cosf(angle);
+            imag_sum -= buffer[n] * sinf(angle);
         }
-        output[k] = std::sqrt(real_sum * real_sum + imag_sum * imag_sum);
+        output[k] = sqrtf(real_sum * real_sum + imag_sum * imag_sum);
     }
 }