We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f1ff4d1 commit 979ecacCopy full SHA for 979ecac
torchvision/csrc/ops/cuda/deform_conv2d_kernel.cu
@@ -230,7 +230,8 @@ void deformable_im2col(
230
at::Tensor data_col) {
231
at::cuda::CUDAGuard device_guard(input.get_device());
232
233
- const int64_t num_kernels = (int64_t)n_in_channels * out_h * out_w * parallel_imgs;
+ const int64_t num_kernels =
234
+ (int64_t)n_in_channels * out_h * out_w * parallel_imgs;
235
236
const unsigned int threads = GET_THREADS();
237
const unsigned int blocks = GET_BLOCKS(threads, num_kernels);
0 commit comments