Skip to content

GitLab

Commit 6c560290 authored by peterjc123's avatar peterjc123 Committed by Francisco Massa
Browse files

Enable symbol annotations for Windows (#1081)

parent 55088157
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
setup.py
View file @ 6c560290
...@@ -115,6 +115,9 @@ def get_extensions(): ...@@ -115,6 +115,9 @@ def get_extensions():
'nvcc': nvcc_flags, 'nvcc': nvcc_flags,
} }
if sys.platform == 'win32':
define_macros += [('torchvision_EXPORTS', None)]
sources = [os.path.join(extensions_dir, s) for s in sources] sources = [os.path.join(extensions_dir, s) for s in sources]
include_dirs = [extensions_dir] include_dirs = [extensions_dir]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment