Skip to content

Commit 3010ff2

Browse files
Address internal linters (#9201)
Co-authored-by: Antoine Simoulin <[email protected]>
1 parent cdc1fee commit 3010ff2

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

torchvision/csrc/ops/cuda/nms_kernel.cu

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff 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[];

0 commit comments

Comments
 (0)