From 4ee9e4b6a2a4dc4dfebfec3c382ff867c3351e3d Mon Sep 17 00:00:00 2001
From: peter <peter_jiachen@163.com>
Date: Wed, 3 Jul 2019 19:59:31 +0800
Subject: [PATCH] Enable symbol annotations for Windows

---
 setup.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/setup.py b/setup.py
index 5e87a7a099a..7c4c446f8ee 100644
--- a/setup.py
+++ b/setup.py
@@ -115,6 +115,9 @@ def get_extensions():
             'nvcc': nvcc_flags,
         }
 
+    if sys.platform == 'win32':
+        define_macros += [('torchvision_EXPORTS', None)]
+
     sources = [os.path.join(extensions_dir, s) for s in sources]
 
     include_dirs = [extensions_dir]