[Fix] Fix DCN TensorRT plugin (#2408)

mmcv/ops/csrc/tensorrt/plugins/trt_deform_conv.cpp

@@ -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];
     }
 

mmcv/ops/csrc/tensorrt/plugins/trt_modulated_deform_conv.cpp

@@ -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];
     }
 

mmcv/ops/csrc/tensorrt/plugins/trt_modulated_deform_conv_kernel.cu

@@ -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 =