From 33b3ffd48d4d0d266a9ff4ff26a0e1ce3a6bc7d2 Mon Sep 17 00:00:00 2001 From: Daniel Delgado Vargas Date: Mon, 6 Apr 2026 20:56:21 +0000 Subject: [PATCH] fix: ambiguous min in nms_kernel function --- torchvision/csrc/ops/cuda/nms_kernel.cu | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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(),