fix lint

open-mmlab · Mar 3, 2022 · d3f9a6c · d3f9a6c
1 parent f00bc33
commit d3f9a6c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 7 deletions.
diff --git a/csrc/backend_ops/tensorrt/deform_conv/trt_deform_conv_kernel.cu b/csrc/backend_ops/tensorrt/deform_conv/trt_deform_conv_kernel.cu
@@ -162,4 +162,4 @@ template void deform_conv<float>(const float* input, const float* weight, const
                                  int inputHeight, int inputWidth, int nOutputPlane, int kW, int kH,
                                  int dW, int dH, int padW, int padH, int dilationW, int dilationH,
                                  int group, int deformable_group, int im2col_step,
-                                 cublasHandle_t cublas_handle, cudaStream_t stream);
+                                 cublasHandle_t cublas_handle, cudaStream_t stream);
diff --git a/csrc/backend_ops/tensorrt/deform_conv/trt_deform_conv_kernel.hpp b/csrc/backend_ops/tensorrt/deform_conv/trt_deform_conv_kernel.hpp
@@ -17,4 +17,4 @@ void deform_conv(const scalar_t* input, const scalar_t* weight, const scalar_t*
                  int inputWidth, int nOutputPlane, int kW, int kH, int dW, int dH, int padW,
                  int padH, int dilationW, int dilationH, int group, int deformable_group,
                  int im2col_step, cublasHandle_t cublas_handle, cudaStream_t stream);
-#endif  // TRT_DEFORM_CONV_KERNEL_HPP
+#endif  // TRT_DEFORM_CONV_KERNEL_HPP
diff --git a/mmdeploy/mmcv/ops/deform_conv.py b/mmdeploy/mmcv/ops/deform_conv.py
@@ -17,12 +17,8 @@ def deform_conv__default(ctx,
                          bias=False,
                          im2col_step=32):
     """Rewrite symbolic function for default backend."""
-    assert not bias, 'The "bias" parameter should be False.'
-    assert groups == 1, 'The "groups" parameter should be 1.'
-    domain = 'mmdeploy'
-    op_name = 'MMCVDeformConv2d'
     return g.op(
-        f'{domain}::{op_name}',
+        'mmdeploy::MMCVDeformConv2d',
         input,
         offset,
         weight,