Unverified Commit 2f4efd3f authored by Francisco Massa's avatar Francisco Massa Committed by GitHub
Browse files

Revert "Added "-fopenmp" cflags (#2783) (#3006)" (#3038)

This reverts commit 74de51d6.
parent 7ffc9520
......@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment