diff --git a/mmcv/ops/points_in_boxes.py b/mmcv/ops/points_in_boxes.py index 50a085cf46..3069867886 100644 --- a/mmcv/ops/points_in_boxes.py +++ b/mmcv/ops/points_in_boxes.py @@ -133,8 +133,6 @@ def points_in_boxes_all(points: Tensor, boxes: Tensor) -> Tensor: if points.device.type != 'npu': if torch.cuda.current_device() != points_device: torch.cuda.set_device(points_device) - elif points.device.type == 'npu': - boxes[:, :, 2] += boxes[:, :, 5] / 2.0 ext_module.points_in_boxes_all_forward(boxes.contiguous(), points.contiguous(), box_idxs_of_pts)