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

This reverts commit 74de51d.
pytorch · Feb 22, 2021 · 2f4efd3 · 2f4efd3
1 parent 7ffc952
commit 2f4efd3
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/setup.py b/setup.py
@@ -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: