Pass thread count to ffmpeg's filtergraph (#209)

pytorch · Sep 10, 2024 · ec24944 · ec24944
1 parent 59cc022
commit ec24944
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/torchcodec/decoders/_core/VideoDecoder.cpp b/src/torchcodec/decoders/_core/VideoDecoder.cpp
@@ -254,6 +254,9 @@ void VideoDecoder::initializeFilterGraphForStream(
   }
   filterState.filterGraph.reset(avfilter_graph_alloc());
   TORCH_CHECK(filterState.filterGraph.get() != nullptr);
+  if (options.ffmpegThreadCount.has_value()) {
+    filterState.filterGraph->nb_threads = options.ffmpegThreadCount.value();
+  }
   const AVFilter* buffersrc = avfilter_get_by_name("buffer");
   const AVFilter* buffersink = avfilter_get_by_name("buffersink");
   enum AVPixelFormat pix_fmts[] = {AV_PIX_FMT_RGB24, AV_PIX_FMT_NONE};