diff --git a/torchvision/csrc/ops/cuda/deform_conv2d_kernel.cu b/torchvision/csrc/ops/cuda/deform_conv2d_kernel.cu index f1b0f75b008..6389d80117f 100644 --- a/torchvision/csrc/ops/cuda/deform_conv2d_kernel.cu +++ b/torchvision/csrc/ops/cuda/deform_conv2d_kernel.cu @@ -230,7 +230,8 @@ void deformable_im2col( at::Tensor data_col) { at::cuda::CUDAGuard device_guard(input.get_device()); - const int64_t num_kernels = (int64_t)n_in_channels * out_h * out_w * parallel_imgs; + const int64_t num_kernels = + (int64_t)n_in_channels * out_h * out_w * parallel_imgs; const unsigned int threads = GET_THREADS(); const unsigned int blocks = GET_BLOCKS(threads, num_kernels);