fix msda

Update chamfer_distance_npu.cpp
momo609 · Aug 5, 2024 · a6d4f0d · a6d4f0d
1 parent 6728a04
commit a6d4f0d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 10 deletions.
diff --git a/mmcv/ops/csrc/pytorch/npu/chamfer_distance_npu.cpp b/mmcv/ops/csrc/pytorch/npu/chamfer_distance_npu.cpp
@@ -35,7 +35,6 @@ void chamfer_distance_forward_npu(Tensor XYZ1, Tensor XYZ2, Tensor dist1,
   dist2.copy_(distf2);
 }
 
-
 void chamfer_distance_backward_npu(Tensor xyz1, Tensor xyz2, Tensor idx1,
                                    Tensor idx2, Tensor grad_dist1,
                                    Tensor grad_dist2, Tensor grad_xyz1,

diff --git a/mmcv/ops/csrc/pytorch/npu/ms_deform_attn_npu.cpp b/mmcv/ops/csrc/pytorch/npu/ms_deform_attn_npu.cpp
@@ -57,15 +57,9 @@ Tensor ms_deform_attn_forward_npu(const Tensor &value,
       value.size(0), sampling_locations.size(1), value.size(2) * value.size(3)};
   at::Tensor output = at::zeros(output_size, value_fp32.options());
 
-  OpCommand cmd;
-  cmd.Name("MultiScaleDeformableAttnFunction")
-      .Input(value_fp32)
-      .Input(value_spatial_shapes_int32)
-      .Input(value_level_start_index_int32)
-      .Input(sampling_locations_fp32)
-      .Input(attention_weights_fp32)
-      .Output(output)
-      .Run();
+  EXEC_NPU_CMD(aclnnMultiScaleDeformableAttnFunction, value_fp32,
+               value_spatial_shapes_int32, value_level_start_index_int32,
+               sampling_locations_fp32, attention_weights_fp32, output);
 
   at::Tensor real_output = output;
   if (value.scalar_type() != at::kFloat) {