File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
torchvision/csrc/ops/cuda Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -34,8 +34,8 @@ __global__ void nms_kernel_impl(
3434 double iou_threshold,
3535 const T* dev_boxes,
3636 unsigned long long * dev_mask) {
37- const int row_start = blockIdx .y ;
38- const int col_start = blockIdx .x ;
37+ const auto row_start = blockIdx .y ;
38+ const auto col_start = blockIdx .x ;
3939
4040 if (row_start > col_start)
4141 return ;
@@ -59,7 +59,7 @@ __global__ void nms_kernel_impl(
5959 __syncthreads ();
6060
6161 if (threadIdx .x < row_size) {
62- const int cur_box_idx = threadsPerBlock * row_start + threadIdx .x ;
62+ const auto cur_box_idx = threadsPerBlock * row_start + threadIdx .x ;
6363 const T* cur_box = dev_boxes + cur_box_idx * 4 ;
6464 int i = 0 ;
6565 unsigned long long t = 0 ;
@@ -84,7 +84,7 @@ __global__ static void gather_keep_from_mask(
8484 // Taken and adapted from mmcv
8585 // https://github.com/open-mmlab/mmcv/blob/03ce9208d18c0a63d7ffa087ea1c2f5661f2441a/mmcv/ops/csrc/common/cuda/nms_cuda_kernel.cuh#L76
8686 const int col_blocks = ceil_div (n_boxes, threadsPerBlock);
87- const int thread_id = threadIdx .x ;
87+ const auto thread_id = threadIdx .x ;
8888
8989 // Mark the bboxes which have been removed.
9090 extern __shared__ unsigned long long removed[];
You can’t perform that action at this time.
0 commit comments