begin work on pad

Author: justincdavis

test/test_transforms_v2.py

@@ -4728,6 +4728,7 @@ def test_kernel_video(self):
             make_segmentation_mask,
             make_video,
             make_keypoints,
+            make_image_cvcuda,
         ],
     )
     def test_functional(self, make_input):
@@ -4746,9 +4747,16 @@ def test_functional(self, make_input):
             (F.pad_bounding_boxes, tv_tensors.BoundingBoxes),
             (F.pad_mask, tv_tensors.Mask),
             (F.pad_video, tv_tensors.Video),
+            pytest.param(
+                F.pad_image_cvcuda,
+                "cvcuda.Tensor",
+                marks=pytest.mark.skipif(not CVCUDA_AVAILABLE, reason="test requires CVCUDA"),
+            ),
         ],
     )
     def test_functional_signature(self, kernel, input_type):
+        if input_type == "cvcuda.Tensor":
+            input_type = _import_cvcuda().Tensor
         check_functional_kernel_signature_match(F.pad, kernel=kernel, input_type=input_type)
 
     @pytest.mark.parametrize(
@@ -4761,6 +4769,7 @@ def test_functional_signature(self, kernel, input_type):
             make_segmentation_mask,
             make_video,
             make_keypoints,
+            make_image_cvcuda,
         ],
     )
     def test_transform(self, make_input):

torchvision/transforms/v2/functional/_geometry.py

@@ -1682,6 +1682,49 @@ def _pad_with_vector_fill(
 _pad_image_pil = _register_kernel_internal(pad, PIL.Image.Image)(_FP.pad)
 
 
+if _CVCUDA_AVAILABLE:
+    cvcuda = _import_cvcuda()
+    _pad_mode_to_cvcuda = {
+        "constant": cvcuda.BorderType.CONSTANT,
+        "reflect": cvcuda.BorderType.REFLECT,
+        "replicate": cvcuda.BorderType.REPLICATE,
+        "symmetric": cvcuda.BorderType.WRAP,
+    }
+
+
+def _pad_cvcuda(
+    image: "cvcuda.Tensor",
+    padding: list[int],
+    fill: Optional[Union[int, float, list[float]]] = None,
+    padding_mode: str = "constant",
+) -> "cvcuda.Tensor":
+    cvcuda = _import_cvcuda()
+
+    if padding_mode not in _pad_mode_to_cvcuda:
+        raise ValueError(f"Padding mode '{padding_mode}' is not supported with CVCUDA")
+
+    if fill is None:
+        fill = 0
+    if isinstance(fill, (int, float)):
+        fill = [fill] * image.shape[3]
+
+    left, right, top, bottom = _parse_pad_padding(padding)
+
+    return cvcuda.copymakeborder(
+        image,
+        border_mode=_pad_mode_to_cvcuda[padding_mode],
+        border_value=fill,
+        top=top,
+        left=left,
+        bottom=bottom,
+        right=right,
+    )
+
+
+if _CVCUDA_AVAILABLE:
+    _register_kernel_internal(pad, _import_cvcuda().Tensor)(_pad_cvcuda)
+
+
 @_register_kernel_internal(pad, tv_tensors.Mask)
 def pad_mask(
     mask: torch.Tensor,