Fix bug when including headers (#1052)

15e24bdf · buoyancy99 · Soumith Chintala · 44c66270 · 15e24bdf
Commit 15e24bdf authored Jun 25, 2019 by buoyancy99 Committed by Soumith Chintala Jun 25, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/csrc/nms.h torchvision/csrc/nms.h +1 -1

No files found.
--- a/torchvision/csrc/nms.h
+++ b/torchvision/csrc/nms.h
@@ -2,7 +2,7 @@
 #include "cpu/vision_cpu.h"

 #ifdef WITH_CUDA
-#include "cuda/vision_cpu.h"
+#include "cuda/vision_cuda.h"
 #endif

 at::Tensor nms(