Skip to content

GitLab

Unverified Commit 89253699 authored by Derry Lin's avatar Derry Lin Committed by GitHub
Browse files

[Fix] Fix DCN TensorRT plugin (#2408)

parent 652b1bf2
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
mmcv/ops/csrc/tensorrt/plugins/trt_deform_conv.cpp
View file @ 89253699
...@@ -286,7 +286,7 @@ nvinfer1::IPluginV2 *DeformableConvPluginDynamicCreator::createPlugin( ...@@ -286,7 +286,7 @@ nvinfer1::IPluginV2 *DeformableConvPluginDynamicCreator::createPlugin(
deformableGroup = static_cast<const int *>(fc->fields[i].data)[0]; deformableGroup = static_cast<const int *>(fc->fields[i].data)[0];
} }
if (field_name.compare("group") == 0) { if (field_name.compare("groups") == 0) {
group = static_cast<const int *>(fc->fields[i].data)[0]; group = static_cast<const int *>(fc->fields[i].data)[0];
} }
......
mmcv/ops/csrc/tensorrt/plugins/trt_modulated_deform_conv.cpp
View file @ 89253699
...@@ -258,7 +258,7 @@ nvinfer1::IPluginV2 *ModulatedDeformableConvPluginDynamicCreator::createPlugin( ...@@ -258,7 +258,7 @@ nvinfer1::IPluginV2 *ModulatedDeformableConvPluginDynamicCreator::createPlugin(
deformableGroup = static_cast<const int *>(fc->fields[i].data)[0]; deformableGroup = static_cast<const int *>(fc->fields[i].data)[0];
} }
if (field_name.compare("group") == 0) { if (field_name.compare("groups") == 0) {
group = static_cast<const int *>(fc->fields[i].data)[0]; group = static_cast<const int *>(fc->fields[i].data)[0];
} }
......
mmcv/ops/csrc/tensorrt/plugins/trt_modulated_deform_conv_kernel.cu
View file @ 89253699
...@@ -75,9 +75,9 @@ void ModulatedDeformConvForwardCUDAKernelLauncher( ...@@ -75,9 +75,9 @@ void ModulatedDeformConvForwardCUDAKernelLauncher(
const size_t input_step = channels * height * width; const size_t input_step = channels * height * width;
const size_t offset_step = const size_t offset_step =
deformable_group * kernel_h * kernel_w * 2 * height * width; deformable_group * kernel_h * kernel_w * 2 * height_out * width_out;
const size_t mask_step = const size_t mask_step =
deformable_group * kernel_h * kernel_w * height * width; deformable_group * kernel_h * kernel_w * height_out * width_out;
const size_t out_step = channels_out * height_out * width_out; const size_t out_step = channels_out * height_out * width_out;
const size_t out_group_step = out_step / group; const size_t out_group_step = out_step / group;
const size_t col_g_step = const size_t col_g_step =
......
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