[Fix] Fix the compilation error on windows (#1657)

open-mmlab · Jan 11, 2022 · ccdc61c · ccdc61c
1 parent b5ec327
commit ccdc61c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/mmcv/ops/csrc/common/cuda/convex_iou_cuda_kernel.cuh b/mmcv/ops/csrc/common/cuda/convex_iou_cuda_kernel.cuh
@@ -10,7 +10,7 @@
 
 #define MAXN 100
 #define NMAX 512
-const double EPS = 1E-8;
+__device__ const double EPS = 1E-8;
 
 __device__ inline int sig(double d) { return (d > EPS) - (d < -EPS); }
 

diff --git a/mmcv/ops/csrc/common/cuda/min_area_polygons_cuda.cuh b/mmcv/ops/csrc/common/cuda/min_area_polygons_cuda.cuh
@@ -9,8 +9,7 @@
 #endif
 
 #define MAXN 20
-const float EPS = 1E-8;
-const float PI = 3.1415926;
+__device__ const float PI = 3.1415926;
 
 struct Point {
   float x, y;