fix chamfer_distance cuda bug len should div to 4

open-mmlab · May 23, 2024 · 14f628d · 14f628d
1 parent 768d6cb
commit 14f628d
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/test_ops/test_chamfer_distance.py b/tests/test_ops/test_chamfer_distance.py
@@ -54,9 +54,7 @@ def torch_to_np_type(dtype):
             not IS_NPU_AVAILABLE, reason='requires NPU support'))
 ])
 @pytest.mark.parametrize('dtype', [torch.half, torch.float32])
-@pytest.mark.parametrize('shape', [(2, 600, 2), (2, 600, 2)],
-                                [(1, 1, 2), (1, 1, 2)],
-                                [(1, 7, 2), (1, 7, 2)])
+@pytest.mark.parametrize('shape', [(2, 600, 2), (1, 1, 2), (7, 7, 2)])
 def test_chamfer_distance_npu_dynamic_shape(dtype, device, shape):
     bs = shape[0]
     ns = shape[1]