windows support

e21ee09b · rusty1s · 609d1f8d · e21ee09b
Commit e21ee09b authored Dec 27, 2018 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

setup.py setup.py +6 -1

No files found.
--- a/setup.py
+++ b/setup.py
+import platform
 from setuptools import setup, find_packages
 import torch
 from torch.utils.cpp_extension import CppExtension, CUDAExtension, CUDA_HOME

+extra_compile_args = []
+if platform.system() != 'Windows':
+    extra_compile_args += ['-Wno-unused-variable']
+
 ext_modules = [
    CppExtension(
        'torch_scatter.scatter_cpu', ['cpu/scatter.cpp'],
-        extra_compile_args=['-Wno-unused-variable'])
+        extra_compile_args=extra_compile_args)
 ]
 cmdclass = {'build_ext': torch.utils.cpp_extension.BuildExtension}