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
f319a874
Commit
f319a874
authored
Sep 09, 2021
by
rusty1s
Browse files
[ci skip]
parent
77f6ba5d
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
1 addition
and
4 deletions
+1
-4
.github/workflows/cuda/cu111-Windows-env.sh
.github/workflows/cuda/cu111-Windows-env.sh
+1
-1
setup.py
setup.py
+0
-3
No files found.
.github/workflows/cuda/cu111-Windows-env.sh
View file @
f319a874
...
...
@@ -5,4 +5,4 @@ PATH=${CUDA_HOME}/bin:$PATH
PATH
=
/c/Program
\
Files
\ \(
x86
\)
/Microsoft
\
Visual
\
Studio/2017/BuildTools/MSBuild/15.0/Bin:
$PATH
export
FORCE_CUDA
=
1
export
TORCH_CUDA_ARCH_LIST
=
"
5
.0+PTX;
6.0;
7.0;7.5;8.0;8.6"
export
TORCH_CUDA_ARCH_LIST
=
"
6
.0+PTX;7.0;7.5;8.0;8.6"
setup.py
View file @
f319a874
...
...
@@ -33,9 +33,6 @@ def get_extensions():
extra_compile_args
=
{
'cxx'
:
[
'-O2'
]}
extra_link_args
=
[
'-s'
]
if
sys
.
platform
==
'win32'
:
extra_compile_args
[
'cxx'
].
append
(
'/Zm500'
)
info
=
parallel_info
()
if
(
'backend: OpenMP'
in
info
and
'OpenMP not found'
not
in
info
and
sys
.
platform
!=
'darwin'
):
...
...
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