add op

open-mmlab · Dec 28, 2023 · 6a1ddd4 · 6a1ddd4
1 parent 62332fc
commit 6a1ddd4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmcv/ops/csrc/pytorch/npu/chamfer_distance_npu.cpp b/mmcv/ops/csrc/pytorch/npu/chamfer_distance_npu.cpp
@@ -4,7 +4,7 @@ using namespace NPU_NAME_SPACE;
 using namespace std;
 
 void chamfer_distance_forward_npu(Tensor XYZ1, Tensor XYZ2, Tensor dist1,
-                                    Tensor dist2, Tensor idx1, Tensor idx2) {
+                                  Tensor dist2, Tensor idx1, Tensor idx2) {
   at::Tensor xyz1 = at::ones_like(XYZ1);
   at::Tensor xyz2 = at::ones_like(XYZ2);
   xyz1 = XYZ1.transpose(1, 2).transpose(0, 1);