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
vision
Commits
2f4efd3f
Unverified
Commit
2f4efd3f
authored
Feb 22, 2021
by
Francisco Massa
Committed by
GitHub
Feb 22, 2021
Browse files
Revert "Added "-fopenmp" cflags (#2783) (#3006)" (#3038)
This reverts commit
74de51d6
.
parent
7ffc9520
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
6 deletions
+7
-6
setup.py
setup.py
+7
-6
No files found.
setup.py
View file @
2f4efd3f
...
...
@@ -181,9 +181,7 @@ def get_extensions():
define_macros
=
[]
extra_compile_args
=
{
'cxx'
:
[]
}
extra_compile_args
=
{}
if
(
torch
.
cuda
.
is_available
()
and
((
CUDA_HOME
is
not
None
)
or
is_rocm_pytorch
))
\
or
os
.
getenv
(
'FORCE_CUDA'
,
'0'
)
==
'1'
:
extension
=
CUDAExtension
...
...
@@ -198,13 +196,16 @@ def get_extensions():
else
:
define_macros
+=
[(
'WITH_HIP'
,
None
)]
nvcc_flags
=
[]
extra_compile_args
[
'nvcc'
]
=
nvcc_flags
extra_compile_args
=
{
'cxx'
:
[],
'nvcc'
:
nvcc_flags
,
}
if
sys
.
platform
==
'win32'
:
define_macros
+=
[(
'torchvision_EXPORTS'
,
None
)]
extra_compile_args
.
setdefault
(
'cxx'
,
[])
extra_compile_args
[
'cxx'
].
append
(
'/MP'
)
elif
sys
.
platform
==
'linux'
:
extra_compile_args
[
'cxx'
].
append
(
'-fopenmp'
)
debug_mode
=
os
.
getenv
(
'DEBUG'
,
'0'
)
==
'1'
if
debug_mode
:
...
...
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