Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
torch-sparse
Commits
bf040066
Commit
bf040066
authored
Feb 22, 2020
by
rusty1s
Browse files
refix library
parent
55b83f67
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
1 deletion
+7
-1
setup.py
setup.py
+7
-1
No files found.
setup.py
View file @
bf040066
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
]
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment