diff --git a/torchvision/csrc/io/image/cpu/jpegcommon.h b/torchvision/csrc/io/image/cpu/jpegcommon.h index bbfe6084457..3131da589f6 100644 --- a/torchvision/csrc/io/image/cpu/jpegcommon.h +++ b/torchvision/csrc/io/image/cpu/jpegcommon.h @@ -1,6 +1,8 @@ #pragma once #if JPEG_FOUND +#include + #include #include #endif diff --git a/torchvision/csrc/io/image/cpu/writejpeg_impl.cpp b/torchvision/csrc/io/image/cpu/writejpeg_impl.cpp index 06941162087..ab19ee0a854 100644 --- a/torchvision/csrc/io/image/cpu/writejpeg_impl.cpp +++ b/torchvision/csrc/io/image/cpu/writejpeg_impl.cpp @@ -1,4 +1,5 @@ #include "writejpeg_impl.h" + #include "jpegcommon.h" namespace vision { diff --git a/torchvision/csrc/io/image/cpu/writepng_impl.cpp b/torchvision/csrc/io/image/cpu/writepng_impl.cpp index e7a8bf96ad5..62b66fd9852 100644 --- a/torchvision/csrc/io/image/cpu/writepng_impl.cpp +++ b/torchvision/csrc/io/image/cpu/writepng_impl.cpp @@ -1,4 +1,5 @@ #include "writejpeg_impl.h" + #include "pngcommon.h" namespace vision {