Skip to content

GitLab

Commit 15e24bdf authored by buoyancy99's avatar buoyancy99 Committed by Soumith Chintala
Browse files

Fix bug when including headers (#1052)

parent 44c66270
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/csrc/nms.h
View file @ 15e24bdf
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
#include "cpu/vision_cpu.h" #include "cpu/vision_cpu.h"
#ifdef WITH_CUDA #ifdef WITH_CUDA
#include "cuda/vision_cpu.h" #include "cuda/vision_cuda.h"
#endif #endif
at::Tensor nms( at::Tensor nms(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment