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
26823fc5
Commit
26823fc5
authored
Sep 08, 2021
by
rusty1s
Browse files
[ci skip]
parent
3f6db832
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
1 addition
and
3 deletions
+1
-3
.github/workflows/cuda/cu111-Windows-env.sh
.github/workflows/cuda/cu111-Windows-env.sh
+1
-1
setup.py
setup.py
+0
-2
No files found.
.github/workflows/cuda/cu111-Windows-env.sh
View file @
26823fc5
...
...
@@ -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
=
"
3.5;
5.0+PTX;6.0;7.0;7.5;8.0;8.6"
export
TORCH_CUDA_ARCH_LIST
=
"5.0+PTX;6.0;7.0;7.5;8.0;8.6"
setup.py
View file @
26823fc5
...
...
@@ -49,8 +49,6 @@ def get_extensions():
nvcc_flags
=
os
.
getenv
(
'NVCC_FLAGS'
,
''
)
nvcc_flags
=
[]
if
nvcc_flags
==
''
else
nvcc_flags
.
split
(
' '
)
nvcc_flags
+=
[
'--expt-relaxed-constexpr'
,
'-O2'
]
if
sys
.
platform
==
'win32'
:
nvcc_flags
+=
[
'/Zm200'
]
extra_compile_args
[
'nvcc'
]
=
nvcc_flags
name
=
main
.
split
(
os
.
sep
)[
-
1
][:
-
4
]
...
...
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