fix: windows install

Aziks0 · Apr 1, 2022 · f3a5981 · f3a5981
1 parent 6886f85
commit f3a5981
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/detectron2/layers/csrc/nms_rotated/nms_rotated_cuda.cu b/detectron2/layers/csrc/nms_rotated/nms_rotated_cuda.cu
@@ -3,13 +3,15 @@
 #include <ATen/cuda/CUDAContext.h>
 #include <c10/cuda/CUDAGuard.h>
 #include <ATen/cuda/CUDAApplyUtils.cuh>
+/*
 #ifdef WITH_CUDA
 #include "../box_iou_rotated/box_iou_rotated_utils.h"
 #endif
 // TODO avoid this when pytorch supports "same directory" hipification
 #ifdef WITH_HIP
-#include "box_iou_rotated/box_iou_rotated_utils.h"
 #endif
+*/
+#include "box_iou_rotated/box_iou_rotated_utils.h"
 
 using namespace detectron2;