fix

open-mmlab · May 6, 2024 · ab19812 · ab19812
1 parent be40170
commit ab19812
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmcv/ops/csrc/pytorch/npu/ms_deform_attn_npu.cpp b/mmcv/ops/csrc/pytorch/npu/ms_deform_attn_npu.cpp
@@ -91,8 +91,8 @@ void ms_deform_attn_backward_npu(const Tensor &value, const Tensor &spatial_shap
                                 Tensor &grad_sampling_loc,
                                 Tensor &grad_attn_weight, const int im2col_step) {
   check_support(value, attn_weight);
-  EXEC_NPU_CMD(aclnnMultiScaleDeformableAttentionGrad, value, spatial_shapes, level_start_index,
-               sampling_loc, attn_weight, grad_output
+  EXEC_NPU_CMD(aclnnMultiScaleDeformableAttentionGrad, value, spatial_shapes,
+               level_start_index, sampling_loc, attn_weight, grad_output,
                grad_value, grad_sampling_loc, grad_attn_weight);
 }