Skip to content

Commit 09ce675

Browse files
fmassadatumbox
andauthored
Remove unneded ifdef in nms_kernel.cu (#3326)
Co-authored-by: Vasilis Vryniotis <[email protected]>
1 parent 3673716 commit 09ce675

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

torchvision/csrc/ops/cuda/nms_kernel.cu

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -101,11 +101,7 @@ at::Tensor nms_kernel(
101101
" and ",
102102
scores.size(0))
103103

104-
#if defined(WITH_CUDA) || defined(WITH_HIP)
105104
at::cuda::CUDAGuard device_guard(dets.device());
106-
#else
107-
TORCH_CHECK(false, "Not compiled with GPU support");
108-
#endif
109105

110106
if (dets.numel() == 0) {
111107
return at::empty({0}, dets.options().dtype(at::kLong));

0 commit comments

Comments
 (0)