Skip to content

Commit

Permalink
fix points_in_box bug.
Browse files Browse the repository at this point in the history
  • Loading branch information
momo609 committed Jun 11, 2024
1 parent 316ccbc commit 4d5efef
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions mmcv/ops/csrc/pytorch/npu/three_interpolate_npu.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -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();
Expand Down

0 comments on commit 4d5efef

Please sign in to comment.