update rotate to main standards

Author: justincdavis

test/test_transforms_v2.py

@@ -25,7 +25,6 @@
     assert_equal,
     cache,
     cpu_and_cuda,
-    cvcuda_to_pil_compatible_tensor,
     freeze_rng_state,
     ignore_jit_no_profile_information_warning,
     make_bounding_boxes,
@@ -2149,14 +2148,14 @@ def test_functional(self, make_input):
             (F.rotate_video, tv_tensors.Video),
             (F.rotate_keypoints, tv_tensors.KeyPoints),
             pytest.param(
-                F._geometry._rotate_cvcuda,
-                "cvcuda.Tensor",
+                F._geometry._rotate_image_cvcuda,
+                None,
                 marks=pytest.mark.skipif(not CVCUDA_AVAILABLE, reason="CVCUDA not available"),
             ),
         ],
     )
     def test_functional_signature(self, kernel, input_type):
-        if input_type == "cvcuda.Tensor":
+        if kernel is F._geometry._rotate_image_cvcuda:
             input_type = _import_cvcuda().Tensor
         check_functional_kernel_signature_match(F.rotate, kernel=kernel, input_type=input_type)
 
@@ -2205,8 +2204,8 @@ def test_functional_image_correctness(self, angle, center, interpolation, expand
         actual = F.rotate(image, angle=angle, center=center, interpolation=interpolation, expand=expand, fill=fill)
 
         if make_input is make_image_cvcuda:
-            actual = cvcuda_to_pil_compatible_tensor(actual)
-            image = cvcuda_to_pil_compatible_tensor(image)
+            actual = F.cvcuda_to_tensor(actual)[0].cpu()
+            image = F.cvcuda_to_tensor(image)[0].cpu()
 
         expected = F.to_image(
             F.rotate(
@@ -2256,8 +2255,8 @@ def test_transform_image_correctness(self, center, interpolation, expand, fill,
         torch.manual_seed(seed)
 
         if make_input is make_image_cvcuda:
-            actual = cvcuda_to_pil_compatible_tensor(actual)
-            image = cvcuda_to_pil_compatible_tensor(image)
+            actual = F.cvcuda_to_tensor(actual)[0].cpu()
+            image = F.cvcuda_to_tensor(image)[0].cpu()
 
         expected = F.to_image(transform(F.to_pil_image(image)))
 

torchvision/transforms/v2/_geometry.py

@@ -29,7 +29,6 @@
     is_pure_tensor,
     query_size,
 )
-from .functional._utils import is_cvcuda_tensor
 
 CVCUDA_AVAILABLE = _is_cvcuda_available()
 
@@ -607,7 +606,8 @@ class RandomRotation(Transform):
 
     _v1_transform_cls = _transforms.RandomRotation
 
-    _transformed_types = Transform._transformed_types + (is_cvcuda_tensor,)
+    if CVCUDA_AVAILABLE:
+        _transformed_types = Transform._transformed_types + (_is_cvcuda_tensor,)
 
     def __init__(
         self,

torchvision/transforms/v2/functional/_geometry.py

@@ -1559,7 +1559,7 @@ def rotate_video(
     }
 
 
-def _rotate_cvcuda(
+def _rotate_image_cvcuda(
     inpt: "cvcuda.Tensor",
     angle: float,
     interpolation: Union[InterpolationMode, int] = InterpolationMode.NEAREST,
@@ -1642,7 +1642,7 @@ def _rotate_cvcuda(
 
 
 if CVCUDA_AVAILABLE:
-    _register_kernel_internal(rotate, _import_cvcuda().Tensor)(_rotate_cvcuda)
+    _register_kernel_internal(rotate, _import_cvcuda().Tensor)(_rotate_image_cvcuda)
 
 
 def pad(