Fixing headers.

pytorch · Jan 27, 2021 · 8450858 · 8450858
1 parent 7fdc7ba
commit 8450858
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git 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 <cstddef>
+
 #include <jpeglib.h>
 #include <setjmp.h>
 #endif

diff --git 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
@@ -1,4 +1,5 @@
 #include "writejpeg_impl.h"
+
 #include "pngcommon.h"
 
 namespace vision {