Commit bf040066 authored by rusty1s's avatar rusty1s
Browse files

refix library

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