Skip to content

GitLab

Unverified Commit ece32796 authored by ychan's avatar ychan Committed by GitHub
Browse files

fix dcn forward bug (#562)

parent 779f47ba
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
mmcv/ops/csrc/parrots/deform_conv_cuda.cu
View file @ ece32796
...@@ -267,6 +267,7 @@ void DeformConvForwardCUDAKernelLauncher( ...@@ -267,6 +267,7 @@ void DeformConvForwardCUDAKernelLauncher(
auto columns_g = columns[g]; auto columns_g = columns[g];
gemm(ctx, 1, false, weight_g, false, columns_g, 1, output_g); gemm(ctx, 1, false, weight_g, false, columns_g, 1, output_g);
} }
columns = columns.view({columns.dim(0) * columns.dim(1), columns.dim(2)});
} }
output_buffer = output_buffer.view( output_buffer = output_buffer.view(
......
mmcv/ops/csrc/pytorch/deform_conv_cuda.cu
View file @ ece32796
...@@ -276,6 +276,8 @@ void DeformConvForwardCUDAKernelLauncher(Tensor input, Tensor weight, ...@@ -276,6 +276,8 @@ void DeformConvForwardCUDAKernelLauncher(Tensor input, Tensor weight,
.addmm_(weight[g].flatten(1), columns[g]) .addmm_(weight[g].flatten(1), columns[g])
.view_as(output_buffer[elt][g]); .view_as(output_buffer[elt][g]);
} }
columns =
columns.view({columns.size(0) * columns.size(1), columns.size(2)});
} }
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