Commit bf040066 authored by rusty1s's avatar rusty1s
Browse files

refix library

parent 55b83f67
import os import os
import os.path as osp import os.path as osp
import sys
import glob import glob
from setuptools import setup, find_packages from setuptools import setup, find_packages
...@@ -30,6 +31,11 @@ def get_extensions(): ...@@ -30,6 +31,11 @@ def get_extensions():
nvcc_flags += ['-arch=sm_35', '--expt-relaxed-constexpr'] nvcc_flags += ['-arch=sm_35', '--expt-relaxed-constexpr']
extra_compile_args['nvcc'] = nvcc_flags extra_compile_args['nvcc'] = nvcc_flags
if sys.platform == 'win32':
extra_link_args = ['cusparse.lib']
else:
extra_link_args = ['-lcusparse', '-l', 'cusparse']
extensions_dir = osp.join(osp.dirname(osp.abspath(__file__)), 'csrc') extensions_dir = osp.join(osp.dirname(osp.abspath(__file__)), 'csrc')
main_files = glob.glob(osp.join(extensions_dir, '*.cpp')) main_files = glob.glob(osp.join(extensions_dir, '*.cpp'))
extensions = [] extensions = []
...@@ -53,7 +59,7 @@ def get_extensions(): ...@@ -53,7 +59,7 @@ def get_extensions():
define_macros=define_macros, define_macros=define_macros,
extra_compile_args=extra_compile_args, extra_compile_args=extra_compile_args,
extra_link_args=extra_link_args, extra_link_args=extra_link_args,
libraries=['cusparse', 'metis'], libraries=['metis'],
) )
extensions += [extension] extensions += [extension]
......
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