Unverified Commit 09ce6758 authored by Francisco Massa's avatar Francisco Massa Committed by GitHub
Browse files

Remove unneded ifdef in nms_kernel.cu (#3326)


Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
parent 36737160
...@@ -101,11 +101,7 @@ at::Tensor nms_kernel( ...@@ -101,11 +101,7 @@ at::Tensor nms_kernel(
" and ", " and ",
scores.size(0)) scores.size(0))
#if defined(WITH_CUDA) || defined(WITH_HIP)
at::cuda::CUDAGuard device_guard(dets.device()); at::cuda::CUDAGuard device_guard(dets.device());
#else
TORCH_CHECK(false, "Not compiled with GPU support");
#endif
if (dets.numel() == 0) { if (dets.numel() == 0) {
return at::empty({0}, dets.options().dtype(at::kLong)); return at::empty({0}, dets.options().dtype(at::kLong));
......
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