2025-04-09 nightly release (82750c8)

Author: pytorchbot

src/torchcodec/_core/Encoder.cpp

@@ -37,8 +37,6 @@ void validateSampleRate(const AVCodec& avCodec, int sampleRate) {
 
 AudioEncoder::~AudioEncoder() {}
 
-// TODO-ENCODING: disable ffmpeg logs by default
-
 AudioEncoder::AudioEncoder(
     const torch::Tensor wf,
     int sampleRate,
@@ -51,6 +49,8 @@ AudioEncoder::AudioEncoder(
       wf_.dtype());
   TORCH_CHECK(
       wf_.dim() == 2, "waveform must have 2 dimensions, got ", wf_.dim());
+
+  setFFmpegLogLevel();
   AVFormatContext* avFormatContext = nullptr;
   auto status = avformat_alloc_output_context2(
       &avFormatContext, nullptr, nullptr, fileName.data());

src/torchcodec/_core/FFMPEGCommon.cpp

@@ -158,4 +158,38 @@ SwrContext* allocateSwrContext(
   return swrContext;
 }
 
+void setFFmpegLogLevel() {
+  auto logLevel = AV_LOG_QUIET;
+  const char* logLevelEnvPtr = std::getenv("TORCHCODEC_FFMPEG_LOG_LEVEL");
+  if (logLevelEnvPtr != nullptr) {
+    std::string logLevelEnv(logLevelEnvPtr);
+    if (logLevelEnv == "QUIET") {
+      logLevel = AV_LOG_QUIET;
+    } else if (logLevelEnv == "PANIC") {
+      logLevel = AV_LOG_PANIC;
+    } else if (logLevelEnv == "FATAL") {
+      logLevel = AV_LOG_FATAL;
+    } else if (logLevelEnv == "ERROR") {
+      logLevel = AV_LOG_ERROR;
+    } else if (logLevelEnv == "WARNING") {
+      logLevel = AV_LOG_WARNING;
+    } else if (logLevelEnv == "INFO") {
+      logLevel = AV_LOG_INFO;
+    } else if (logLevelEnv == "VERBOSE") {
+      logLevel = AV_LOG_VERBOSE;
+    } else if (logLevelEnv == "DEBUG") {
+      logLevel = AV_LOG_DEBUG;
+    } else if (logLevelEnv == "TRACE") {
+      logLevel = AV_LOG_TRACE;
+    } else {
+      TORCH_CHECK(
+          false,
+          "Invalid TORCHCODEC_FFMPEG_LOG_LEVEL: ",
+          logLevelEnv,
+          ". Use e.g. 'QUIET', 'PANIC', 'VERBOSE', etc.");
+    }
+  }
+  av_log_set_level(logLevel);
+}
+
 } // namespace facebook::torchcodec

src/torchcodec/_core/FFMPEGCommon.h

@@ -168,4 +168,6 @@ SwrContext* allocateSwrContext(
 // Returns true if sws_scale can handle unaligned data.
 bool canSwsScaleHandleUnalignedData();
 
+void setFFmpegLogLevel();
+
 } // namespace facebook::torchcodec

src/torchcodec/_core/SingleStreamDecoder.cpp

@@ -7,7 +7,6 @@
 #include "src/torchcodec/_core/SingleStreamDecoder.h"
 #include <cstdint>
 #include <cstdio>
-#include <cstdlib>
 #include <iostream>
 #include <limits>
 #include <sstream>
@@ -184,39 +183,6 @@ void SingleStreamDecoder::initializeDecoder() {
   initialized_ = true;
 }
 
-void SingleStreamDecoder::setFFmpegLogLevel() {
-  auto logLevel = AV_LOG_QUIET;
-  const char* logLevelEnv = std::getenv("TORCHCODEC_FFMPEG_LOG_LEVEL");
-  if (logLevelEnv != nullptr) {
-    if (std::strcmp(logLevelEnv, "QUIET") == 0) {
-      logLevel = AV_LOG_QUIET;
-    } else if (std::strcmp(logLevelEnv, "PANIC") == 0) {
-      logLevel = AV_LOG_PANIC;
-    } else if (std::strcmp(logLevelEnv, "FATAL") == 0) {
-      logLevel = AV_LOG_FATAL;
-    } else if (std::strcmp(logLevelEnv, "ERROR") == 0) {
-      logLevel = AV_LOG_ERROR;
-    } else if (std::strcmp(logLevelEnv, "WARNING") == 0) {
-      logLevel = AV_LOG_WARNING;
-    } else if (std::strcmp(logLevelEnv, "INFO") == 0) {
-      logLevel = AV_LOG_INFO;
-    } else if (std::strcmp(logLevelEnv, "VERBOSE") == 0) {
-      logLevel = AV_LOG_VERBOSE;
-    } else if (std::strcmp(logLevelEnv, "DEBUG") == 0) {
-      logLevel = AV_LOG_DEBUG;
-    } else if (std::strcmp(logLevelEnv, "TRACE") == 0) {
-      logLevel = AV_LOG_TRACE;
-    } else {
-      TORCH_CHECK(
-          false,
-          "Invalid TORCHCODEC_FFMPEG_LOG_LEVEL: ",
-          logLevelEnv,
-          ". Use e.g. 'QUIET', 'PANIC', 'VERBOSE', etc.");
-    }
-  }
-  av_log_set_level(logLevel);
-}
-
 int SingleStreamDecoder::getBestStreamIndex(AVMediaType mediaType) {
   AVCodecOnlyUseForCallingAVFindBestStream avCodec = nullptr;
   int streamIndex =

src/torchcodec/_core/SingleStreamDecoder.h

@@ -249,7 +249,6 @@ class SingleStreamDecoder {
   // --------------------------------------------------------------------------
 
   void initializeDecoder();
-  void setFFmpegLogLevel();
   // --------------------------------------------------------------------------
   // DECODING APIS AND RELATED UTILS
   // --------------------------------------------------------------------------

test/test_ops.py

@@ -44,6 +44,7 @@
 from .utils import (
     assert_frames_equal,
     cpu_and_cuda,
+    in_fbcode,
     NASA_AUDIO,
     NASA_AUDIO_MP3,
     NASA_VIDEO,
@@ -1139,6 +1140,7 @@ def test_round_trip(self, tmp_path):
         )
 
     # TODO-ENCODING: test more encoding formats
+    @pytest.mark.skipif(in_fbcode(), reason="TODO: enable ffmpeg CLI")
     @pytest.mark.parametrize("asset", (NASA_AUDIO_MP3, SINE_MONO_S32))
     @pytest.mark.parametrize("bit_rate", (None, 0, 44_100, 999_999_999))
     def test_against_cli(self, asset, bit_rate, tmp_path):