Skip to content

GitLab

Commit 96df2903 authored by rusty1s's avatar rusty1s
Browse files

fix

parent bd7a732a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
setup.py
View file @ 96df2903
...@@ -21,7 +21,7 @@ def get_extensions(): ...@@ -21,7 +21,7 @@ def get_extensions():
Extension = CppExtension Extension = CppExtension
define_macros = [] define_macros = []
extra_compile_args = {'cxx': []} extra_compile_args = {'cxx': []}
extra_link_args = [] extra_link_args = ['metis.lib']
if WITH_CUDA: if WITH_CUDA:
Extension = CUDAExtension Extension = CUDAExtension
...@@ -32,7 +32,7 @@ def get_extensions(): ...@@ -32,7 +32,7 @@ def get_extensions():
extra_compile_args['nvcc'] = nvcc_flags extra_compile_args['nvcc'] = nvcc_flags
if sys.platform == 'win32': if sys.platform == 'win32':
extra_link_args += ['cusparse.lib', 'metis.lib'] extra_link_args += ['cusparse.lib']
else: else:
extra_link_args += ['-lcusparse', '-l', 'cusparse'] extra_link_args += ['-lcusparse', '-l', 'cusparse']
......
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