From 84508588475f97e82ab8c92f0f5be2beacbcc511 Mon Sep 17 00:00:00 2001
From: Vasilis Vryniotis <vvryniotis@fb.com>
Date: Wed, 27 Jan 2021 20:36:26 +0000
Subject: [PATCH] Fixing headers.

---
 torchvision/csrc/io/image/cpu/jpegcommon.h       | 2 ++
 torchvision/csrc/io/image/cpu/writejpeg_impl.cpp | 1 +
 torchvision/csrc/io/image/cpu/writepng_impl.cpp  | 1 +
 3 files changed, 4 insertions(+)

diff --git a/torchvision/csrc/io/image/cpu/jpegcommon.h b/torchvision/csrc/io/image/cpu/jpegcommon.h
index bbfe6084457..293d02fc11e 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 <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
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 {