From 45cc831efd2656725be6d09faf5ba36fe224e613 Mon Sep 17 00:00:00 2001
From: Julian Bouzas <julian.bouzas@collabora.com>
Date: Wed, 23 Sep 2020 13:49:43 -0400
Subject: [PATCH] nvcodec: Add missing CUDAMemory sink caps in h264 and h265
 encoders

Part-of: <https://gitlab.freedesktop.org/gstreamer/gst-plugins-bad/-/merge_requests/1633>
---
 sys/nvcodec/gstnvh264enc.c | 3 ++-
 sys/nvcodec/gstnvh265enc.c | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/sys/nvcodec/gstnvh264enc.c b/sys/nvcodec/gstnvh264enc.c
index a1a584b17c..3c74c87ffc 100644
--- a/sys/nvcodec/gstnvh264enc.c
+++ b/sys/nvcodec/gstnvh264enc.c
@@ -69,7 +69,8 @@ enum
 
 #define DOCUMENTATION_SINK_CAPS \
     "video/x-raw, " DOCUMENTATION_SINK_CAPS_COMM "; " \
-    "video/x-raw(memory:GLMemory), " DOCUMENTATION_SINK_CAPS_COMM
+    "video/x-raw(memory:GLMemory), " DOCUMENTATION_SINK_CAPS_COMM "; " \
+    "video/x-raw(memory:CUDAMemory), " DOCUMENTATION_SINK_CAPS_COMM
 
 #define DOCUMENTATION_SRC_CAPS \
     "video/x-h264, "  \
diff --git a/sys/nvcodec/gstnvh265enc.c b/sys/nvcodec/gstnvh265enc.c
index 6a0dad5f35..df0bdf55d8 100644
--- a/sys/nvcodec/gstnvh265enc.c
+++ b/sys/nvcodec/gstnvh265enc.c
@@ -70,7 +70,8 @@ enum
 
 #define DOCUMENTATION_SINK_CAPS \
     "video/x-raw, " DOCUMENTATION_SINK_CAPS_COMM "; " \
-    "video/x-raw(memory:GLMemory), " DOCUMENTATION_SINK_CAPS_COMM
+    "video/x-raw(memory:GLMemory), " DOCUMENTATION_SINK_CAPS_COMM "; " \
+    "video/x-raw(memory:CUDAMemory), " DOCUMENTATION_SINK_CAPS_COMM
 
 #define DOCUMENTATION_SRC_CAPS \
     "video/x-h265, "  \