Enable symbol annotations for Windows (#1081)

6c560290 · peterjc123 · Francisco Massa · 55088157 · 6c560290
Commit 6c560290 authored Jul 03, 2019 by peterjc123 Committed by Francisco Massa Jul 03, 2019
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

setup.py setup.py +3 -0

No files found.
--- 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]