fix points_in_box bug.

open-mmlab · Jun 11, 2024 · 2333e24 · 2333e24
1 parent c0c6002
commit 2333e24
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmcv/ops/points_in_boxes.py b/mmcv/ops/points_in_boxes.py
@@ -47,7 +47,7 @@ def points_in_boxes_part(points: Tensor, boxes: Tensor) -> Tensor:
     points_device = points.get_device()
     assert points_device == boxes.get_device(), \
         'Points and boxes should be put on the same device'
-    if points_device != 'npu':
+    if points.device.type != 'npu':
         if torch.cuda.current_device() != points_device:
             torch.cuda.set_device(points_device)