Update cpu_da.cpp

deephealthproject · Nov 26, 2019 · fbaaf81 · fbaaf81
1 parent b58232f
commit fbaaf81
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/hardware/cpu/cpu_da.cpp b/src/hardware/cpu/cpu_da.cpp
@@ -8,13 +8,18 @@
 */
 
 
+
 #include <iostream>
 #include <utility>
 #include <cmath>
 
 #include "cpu_hw.h"
 #include "../../random.h"
 
+#ifndef M_PI
+    #define M_PI 3.14159265358979323846
+#endif
+
 void cpu_single_shift(int b, Tensor *A, Tensor *B, vector<int> shift, int mode, float constant){
     for(int c=0; c<B->shape[1]; c++) {
         for(int Bi=0; Bi<B->shape[2];Bi++) {