Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "9c56d88df7c41ea4c3606678d6d3f429897d356d"
Commit 7f914ef7 authored by rusty1s's avatar rusty1s
Browse files

windows cusparse fix

parent d1c42fbb
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
setup.py
View file @ 7f914ef7
...@@ -13,7 +13,7 @@ cmdclass = {} ...@@ -13,7 +13,7 @@ cmdclass = {}
if CUDA_HOME is not None: if CUDA_HOME is not None:
if platform.system() == 'Windows': if platform.system() == 'Windows':
extra_link_args = ['cusparse.lib'], extra_link_args = 'cusparse.lib'
else: else:
extra_link_args = ['-lcusparse', '-l', 'cusparse'], extra_link_args = ['-lcusparse', '-l', 'cusparse'],
...@@ -41,4 +41,5 @@ setup( ...@@ -41,4 +41,5 @@ setup(
tests_require=tests_require, tests_require=tests_require,
ext_modules=ext_modules, ext_modules=ext_modules,
cmdclass=cmdclass, cmdclass=cmdclass,
packages=find_packages(), ) packages=find_packages(),
)
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