diff --git a/mmcv/ops/csrc/pytorch/npu/chamfer_distance_npu.cpp b/mmcv/ops/csrc/pytorch/npu/chamfer_distance_npu.cpp index 2ad0897f316..800e636945d 100644 --- 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);