Skip to content

GitLab

Unverified Commit 4712db75 authored by ZhangShilong's avatar ZhangShilong Committed by GitHub
Browse files

Fix bug of DCN (#819) 

* fix bug of DCN

* fix clang=format
parent a4c37026
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
mmcv/ops/csrc/pytorch/deform_conv_cuda.cu
View file @ 4712db75
...@@ -278,6 +278,8 @@ void DeformConvForwardCUDAKernelLauncher(Tensor input, Tensor weight, ...@@ -278,6 +278,8 @@ void DeformConvForwardCUDAKernelLauncher(Tensor input, Tensor weight,
} }
columns = columns =
columns.view({columns.size(0) * columns.size(1), columns.size(2)}); columns.view({columns.size(0) * columns.size(1), columns.size(2)});
weight = weight.view({weight.size(0) * weight.size(1), weight.size(2),
weight.size(3), weight.size(4)});
} }
output_buffer = output_buffer.view( output_buffer = output_buffer.view(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment