mirrors/gstreamer
mirrors/gstreamer
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gstreamer.git synced 2024-11-23 18:21:04 +00:00
Code Issues 1 Projects Releases Wiki Activity

nvenc: Reduce the number of pre-allocated device memory

Browse source 
The hard-coded upper bound 32 (or 48 depending on resolution) might waste
GPU memory and high resolution encoding causes OUT-OF-MEMORY allocation error
quite easily. This commit calculates the number of required pre-allocated
device memory based on encoding options and it can reduce the amount of device memory
used by nvenc.
This commit is contained in:
Seungha Yang 2019-09-11 14:56:00 +09:00 committed by Matthew Waters
parent 2b7050120e
commit af77988b9f
1 changed files with 34 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
sys/nvcodec/gstnvbaseenc.c
View file

@ -1437,6 +1437,36 @@ gst_nv_base_enc_setup_rate_control (GstNvBaseEnc * nvenc,
} }
} }
static gint
gst_nv_base_enc_calculate_num_prealloc_buffers (GstNvBaseEnc * enc,
NV_ENC_CONFIG * config)
{
gint num_buffers;
/* At least 4 surfaces are required as documented by Nvidia Encoder guide */
num_buffers = 4;
/* + lookahead depth */
num_buffers += config->rcParams.lookaheadDepth;
/* + GOP size */
num_buffers += config->frameIntervalP;
/* hardcoded upper bound "48"
* The worst case
* default num buffers: 4
* maximum allowed lookahead: 32
* max bfraems: 4 -> frameIntervalP: 5
* "4 + 32 + 5" < "48" so it seems to sufficiently safe upper bound */
num_buffers = MIN (num_buffers, 48);
GST_DEBUG_OBJECT (enc, "Calculated num buffers: %d "
"(lookahead %d, frameIntervalP %d)",
num_buffers, config->rcParams.lookaheadDepth, config->frameIntervalP);
return num_buffers;
}
/* GstVideoEncoder::set_format or by nvenc self if new properties were set. /* GstVideoEncoder::set_format or by nvenc self if new properties were set.
* *
* NvEncReconfigureEncoder with following conditions are not allowed * NvEncReconfigureEncoder with following conditions are not allowed
@ -1669,15 +1699,15 @@ gst_nv_base_enc_set_format (GstVideoEncoder * enc, GstVideoCodecState * state)
#if HAVE_NVCODEC_GST_GL #if HAVE_NVCODEC_GST_GL
GstCapsFeatures *features; GstCapsFeatures *features;
#endif #endif
guint num_macroblocks, i; guint i;
guint input_width, input_height; guint input_width, input_height;
input_width = GST_VIDEO_INFO_WIDTH (info); input_width = GST_VIDEO_INFO_WIDTH (info);
input_height = GST_VIDEO_INFO_HEIGHT (info); input_height = GST_VIDEO_INFO_HEIGHT (info);
num_macroblocks = (GST_ROUND_UP_16 (input_width) >> 4) nvenc->n_bufs =
* (GST_ROUND_UP_16 (input_height) >> 4); gst_nv_base_enc_calculate_num_prealloc_buffers (nvenc,
nvenc->n_bufs = (num_macroblocks >= 8160) ? 32 : 48; params->encodeConfig);
/* input buffers */ /* input buffers */
g_array_set_size (nvenc->items, nvenc->n_bufs); g_array_set_size (nvenc->items, nvenc->n_bufs);