update gaussian noise 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,
@@ -4108,7 +4107,7 @@ def test_functional_uint8(self, make_input):
             (F.gaussian_noise_image, tv_tensors.Image),
             (F.gaussian_noise_video, tv_tensors.Video),
             pytest.param(
-                F._misc._gaussian_noise_cvcuda,
+                F._misc._gaussian_noise_image_cvcuda,
                 "cvcuda.Tensor",
                 marks=pytest.mark.skipif(not CVCUDA_AVAILABLE, reason="CV-CUDA not available"),
             ),

torchvision/transforms/v2/_misc.py

@@ -9,6 +9,7 @@
 
 from torchvision import transforms as _transforms, tv_tensors
 from torchvision.transforms.v2 import functional as F, Transform
+from torchvision.transforms.v2.functional._utils import _is_cvcuda_available, _is_cvcuda_tensor
 
 from ._utils import (
     _parse_labels_getter,
@@ -17,10 +18,11 @@
     get_bounding_boxes,
     get_keypoints,
     has_any,
-    is_cvcuda_tensor,
     is_pure_tensor,
 )
 
+CVCUDA_AVAILABLE = _is_cvcuda_available()
+
 
 # TODO: do we want/need to expose this?
 class Identity(Transform):
@@ -241,7 +243,8 @@ class GaussianNoise(Transform):
             Default is True.
     """
 
-    _transformed_types = Transform._transformed_types + (is_cvcuda_tensor,)
+    if CVCUDA_AVAILABLE:
+        _transformed_types = Transform._transformed_types + (_is_cvcuda_tensor,)
 
     def __init__(self, mean: float = 0.0, sigma: float = 0.1, clip=True) -> None:
         super().__init__()

torchvision/transforms/v2/functional/_misc.py

@@ -238,7 +238,7 @@ def _gaussian_noise_pil(
     raise ValueError("Gaussian Noise is not implemented for PIL images.")
 
 
-def _gaussian_noise_cvcuda(
+def _gaussian_noise_image_cvcuda(
     image: "cvcuda.Tensor",
     mean: float = 0.0,
     sigma: float = 0.1,
@@ -264,7 +264,7 @@ def _gaussian_noise_cvcuda(
 
 
 if CVCUDA_AVAILABLE:
-    _register_kernel_internal(gaussian_noise, _import_cvcuda().Tensor)(_gaussian_noise_cvcuda)
+    _register_kernel_internal(gaussian_noise, _import_cvcuda().Tensor)(_gaussian_noise_image_cvcuda)
 
 
 def to_dtype(inpt: torch.Tensor, dtype: torch.dtype = torch.float, scale: bool = False) -> torch.Tensor: