[cuda/c++] PI is needed on host as well, C++20 constants not availabl…

…e yet
alicevision · Jul 31, 2024 · ed66026 · ed66026
1 parent acb7be5
commit ed66026
Showing 1 changed file with 20 additions and 13 deletions.
diff --git a/src/popsift/sift_constants.h b/src/popsift/sift_constants.h
@@ -15,20 +15,27 @@
 #ifndef NINF
 #define NINF              (-INF)
 #endif
-#ifdef M_PI
-#undef M_PI
-// #define M_PI  3.14159265358979323846f
-#endif
-__device__ static const
-float M_PI = 3.14159265358979323846f;
-#ifdef M_PI2
-#undef M_PI2
-// #define M_PI2 (2.0F * M_PI)
-#endif
-__device__ static const
-float M_PI2 = 2.0f * 3.14159265358979323846f;
 
-#define M_4RPI               (4.0f / M_PI)
+#undef USE_CONSTANT_PI
+#ifdef USE_CONSTANT_PI
+  #ifdef M_PI
+    #undef M_PI
+  #endif
+  __device__ static const float M_PI = 3.14159265358979323846f;
+  #ifdef M_PI2
+    #undef M_PI2
+  #endif
+  __device__ static const float M_PI2 = 2.0f * 3.14159265358979323846f;
+  #define M_4RPI               (4.0f / M_PI)
+#else
+  #ifndef M_PI
+    #define M_PI 3.14159265358979323846f
+  #endif
+  #ifndef M_PI2
+    #define M_PI2 (2.0F * M_PI)
+  #endif
+  #define M_4RPI               (4.0f / M_PI)
+#endif
 
 #define DESC_MIN_FLOAT               1E-15F