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-scatter
Commits
3b9a9a90
Commit
3b9a9a90
authored
Jun 16, 2021
by
rusty1s
Browse files
[skip ci] python 3.9 wheels
parent
f9c38ab1
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
3 deletions
+4
-3
.github/workflows/building.yml
.github/workflows/building.yml
+2
-2
setup.py
setup.py
+2
-1
No files found.
.github/workflows/building.yml
View file @
3b9a9a90
...
@@ -10,8 +10,8 @@ jobs:
...
@@ -10,8 +10,8 @@ jobs:
strategy
:
strategy
:
matrix
:
matrix
:
os
:
[
macos-10.15
,
ubuntu-18.04
]
os
:
[
macos-10.15
,
ubuntu-18.04
]
python-version
:
[
3.9
]
python-version
:
[
3.9
]
# [3.6, 3.7, 3.8, 3.9]
torch-version
:
[
1.8.0
]
torch-version
:
[
1.8.0
]
# [1.8.0, 1.9.0]
cuda-version
:
[
'
cpu'
,
'
cu102'
,
'
cu111'
]
cuda-version
:
[
'
cpu'
,
'
cu102'
,
'
cu111'
]
exclude
:
exclude
:
-
os
:
macos-10.15
-
os
:
macos-10.15
...
...
setup.py
View file @
3b9a9a90
...
@@ -34,7 +34,8 @@ def get_extensions():
...
@@ -34,7 +34,8 @@ def get_extensions():
extra_link_args
=
[
'-s'
]
extra_link_args
=
[
'-s'
]
info
=
parallel_info
()
info
=
parallel_info
()
if
'backend: OpenMP'
in
info
and
'OpenMP not found'
not
in
info
:
if
(
'backend: OpenMP'
in
info
and
'OpenMP not found'
not
in
info
and
sys
.
platform
!=
'darwin'
):
extra_compile_args
[
'cxx'
]
+=
[
'-DAT_PARALLEL_OPENMP'
]
extra_compile_args
[
'cxx'
]
+=
[
'-DAT_PARALLEL_OPENMP'
]
if
sys
.
platform
==
'win32'
:
if
sys
.
platform
==
'win32'
:
extra_compile_args
[
'cxx'
]
+=
[
'/openmp'
]
extra_compile_args
[
'cxx'
]
+=
[
'/openmp'
]
...
...
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