mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Fix] Fix DCN TensorRT plugin (#2408)

pull/2415/head
Derry Lin 2022-11-14 19:09:10 +08:00 committed by GitHub
parent 652b1bf207
commit 89253699da
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmcv/ops/csrc/tensorrt/plugins/trt_deform_conv.cpp
View File

@ -286,7 +286,7 @@ nvinfer1::IPluginV2 *DeformableConvPluginDynamicCreator::createPlugin(
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];
}

2
mmcv/ops/csrc/tensorrt/plugins/trt_modulated_deform_conv.cpp
View File

@ -258,7 +258,7 @@ nvinfer1::IPluginV2 *ModulatedDeformableConvPluginDynamicCreator::createPlugin(
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];
}

4
mmcv/ops/csrc/tensorrt/plugins/trt_modulated_deform_conv_kernel.cu
View File

@ -75,9 +75,9 @@ void ModulatedDeformConvForwardCUDAKernelLauncher(
const size_t input_step = channels * height * width;
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 =
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_group_step = out_step / group;
const size_t col_g_step =