diff --git a/mmcv/ops/csrc/pytorch/npu/ms_deform_attn_npu.cpp b/mmcv/ops/csrc/pytorch/npu/ms_deform_attn_npu.cpp index 14c7e30ad0..df23543452 100644 --- a/mmcv/ops/csrc/pytorch/npu/ms_deform_attn_npu.cpp +++ b/mmcv/ops/csrc/pytorch/npu/ms_deform_attn_npu.cpp @@ -59,7 +59,7 @@ Tensor ms_deform_attn_forward_npu(const Tensor &value, EXEC_NPU_CMD(aclnnMultiScaleDeformableAttnFunction, value_fp32, value_spatial_shapes_int32, value_level_start_index_int32, - sampling_locations_fp32, attention_weights_fp32, output) + sampling_locations_fp32, attention_weights_fp32, output); at::Tensor real_output = output; if (value.scalar_type() != at::kFloat) {