fix npu_msda (#3153)

open-mmlab · Jul 24, 2024 · 5916fbd · 5916fbd
1 parent 9f43b8c
commit 5916fbd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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) {