Commit e21ee09b authored by rusty1s's avatar rusty1s
Browse files

windows support

parent 609d1f8d
import platform
from setuptools import setup, find_packages from setuptools import setup, find_packages
import torch import torch
from torch.utils.cpp_extension import CppExtension, CUDAExtension, CUDA_HOME 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 = [ ext_modules = [
CppExtension( CppExtension(
'torch_scatter.scatter_cpu', ['cpu/scatter.cpp'], '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} cmdclass = {'build_ext': torch.utils.cpp_extension.BuildExtension}
......
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