diff --git a/torchvision/csrc/ops/cuda/nms_kernel.cu b/torchvision/csrc/ops/cuda/nms_kernel.cu index 44ce8db6b8e..d5cf35e0c3b 100644 --- a/torchvision/csrc/ops/cuda/nms_kernel.cu +++ b/torchvision/csrc/ops/cuda/nms_kernel.cu @@ -3,6 +3,7 @@ #include #include #include +#include #include "cuda_helpers.h" @@ -188,7 +189,7 @@ at::Tensor nms_kernel( // See https://github.com/pytorch/vision/issues/8713 for more details. gather_keep_from_mask<<< 1, - min(col_blocks, threadsPerBlock), + std::min(col_blocks, threadsPerBlock), col_blocks * sizeof(unsigned long long), stream>>>( keep.data_ptr(),