summaryrefslogtreecommitdiff
path: root/libavutil
diff options
context:
space:
mode:
authorZhao Zhili <zhilizhao@tencent.com>2023-10-27 23:37:00 +0800
committerZhao Zhili <zhilizhao@tencent.com>2023-10-29 13:58:30 +0800
commit63078b45999d116945841146576d914cc0389fe2 (patch)
tree68ec8d7c64364d0c965ce965d5fbd5e64eba85d1 /libavutil
parent891f70c6d55571cc870b42c73b71b60c007e0ded (diff)
avutil/hwcontext_vulkan: cuda doesn't belong to valid_sw_formats
Move it to transfer_get_formats. Signed-off-by: Zhao Zhili <zhilizhao@tencent.com>
Diffstat (limited to 'libavutil')
-rw-r--r--libavutil/hwcontext_vulkan.c25
1 files changed, 13 insertions, 12 deletions
diff --git a/libavutil/hwcontext_vulkan.c b/libavutil/hwcontext_vulkan.c
index 8dd40cb66f..8481427b42 100644
--- a/libavutil/hwcontext_vulkan.c
+++ b/libavutil/hwcontext_vulkan.c
@@ -1664,11 +1664,6 @@ static int vulkan_frames_get_constraints(AVHWDeviceContext *ctx,
NULL, NULL, NULL, NULL, 0, 0) >= 0;
}
-#if CONFIG_CUDA
- if (p->dev_is_nvidia)
- count++;
-#endif
-
constraints->valid_sw_formats = av_malloc_array(count + 1,
sizeof(enum AVPixelFormat));
if (!constraints->valid_sw_formats)
@@ -1684,10 +1679,6 @@ static int vulkan_frames_get_constraints(AVHWDeviceContext *ctx,
}
}
-#if CONFIG_CUDA
- if (p->dev_is_nvidia)
- constraints->valid_sw_formats[count++] = AV_PIX_FMT_CUDA;
-#endif
constraints->valid_sw_formats[count++] = AV_PIX_FMT_NONE;
constraints->min_width = 1;
@@ -2416,12 +2407,22 @@ static int vulkan_transfer_get_formats(AVHWFramesContext *hwfc,
enum AVHWFrameTransferDirection dir,
enum AVPixelFormat **formats)
{
- enum AVPixelFormat *fmts = av_malloc_array(2, sizeof(*fmts));
+ enum AVPixelFormat *fmts;
+ int n = 2;
+
+#if CONFIG_CUDA
+ n++;
+#endif
+ fmts = av_malloc_array(n, sizeof(*fmts));
if (!fmts)
return AVERROR(ENOMEM);
- fmts[0] = hwfc->sw_format;
- fmts[1] = AV_PIX_FMT_NONE;
+ n = 0;
+ fmts[n++] = hwfc->sw_format;
+#if CONFIG_CUDA
+ fmts[n++] = AV_PIX_FMT_CUDA;
+#endif
+ fmts[n++] = AV_PIX_FMT_NONE;
*formats = fmts;
return 0;