[fbsync] Fixed comparison warnings in audio_stream and video_stream (#…

…4007) Summary: * Fixed comparison warnings in audio_stream and video_stream * Fixed clang error Reviewed By: fmassa Differential Revision: D29097721 fbshipit-source-id: c3c11eab92bfcc7c67a9111a75b4a87041bfc655
pytorch · Jun 14, 2021 · bf6c5e9 · bf6c5e9
1 parent 08630d9
commit bf6c5e9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/torchvision/csrc/io/decoder/audio_stream.cpp b/torchvision/csrc/io/decoder/audio_stream.cpp
@@ -7,13 +7,13 @@ namespace ffmpeg {
 
 namespace {
 bool operator==(const AudioFormat& x, const AVFrame& y) {
-  return x.samples == y.sample_rate && x.channels == y.channels &&
-      x.format == y.format;
+  return x.samples == static_cast<size_t>(y.sample_rate) &&
+      x.channels == static_cast<size_t>(y.channels) && x.format == y.format;
 }
 
 bool operator==(const AudioFormat& x, const AVCodecContext& y) {
-  return x.samples == y.sample_rate && x.channels == y.channels &&
-      x.format == y.sample_fmt;
+  return x.samples == static_cast<size_t>(y.sample_rate) &&
+      x.channels == static_cast<size_t>(y.channels) && x.format == y.sample_fmt;
 }
 
 AudioFormat& toAudioFormat(AudioFormat& x, const AVFrame& y) {

diff --git a/torchvision/csrc/io/decoder/video_stream.cpp b/torchvision/csrc/io/decoder/video_stream.cpp
@@ -6,11 +6,13 @@ namespace ffmpeg {
 
 namespace {
 bool operator==(const VideoFormat& x, const AVFrame& y) {
-  return x.width == y.width && x.height == y.height && x.format == y.format;
+  return x.width == static_cast<size_t>(y.width) &&
+      x.height == static_cast<size_t>(y.height) && x.format == y.format;
 }
 
 bool operator==(const VideoFormat& x, const AVCodecContext& y) {
-  return x.width == y.width && x.height == y.height && x.format == y.pix_fmt;
+  return x.width == static_cast<size_t>(y.width) &&
+      x.height == static_cast<size_t>(y.height) && x.format == y.pix_fmt;
 }
 
 VideoFormat& toVideoFormat(VideoFormat& x, const AVFrame& y) {