From 4d5efef4edd2f215bd66d8c0ef9f7fc4bd2be110 Mon Sep 17 00:00:00 2001 From: momo609 <963372609@qq.com> Date: Tue, 11 Jun 2024 15:56:56 +0800 Subject: [PATCH] fix points_in_box bug. --- mmcv/ops/csrc/pytorch/npu/three_interpolate_npu.cpp | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/mmcv/ops/csrc/pytorch/npu/three_interpolate_npu.cpp b/mmcv/ops/csrc/pytorch/npu/three_interpolate_npu.cpp index 28a9c0d71d..f908755478 100644 --- a/mmcv/ops/csrc/pytorch/npu/three_interpolate_npu.cpp +++ b/mmcv/ops/csrc/pytorch/npu/three_interpolate_npu.cpp @@ -37,8 +37,7 @@ void three_interpolate_backward_npu(int b, int c, int n, int m, auto grad_x = at::unsqueeze(grad_out, 3); auto grad_y = at::unsqueeze(grad_points, 3); - EXEC_NPU_CMD(aclnnThreeInterpolateBackward, grad_x, idx, weight, m, - grad_y); + EXEC_NPU_CMD(aclnnThreeInterpolateBackward, grad_x, idx, weight, m, grad_y); auto output = at::squeeze(grad_y, 3); auto res = output.contiguous();