From 2f4efd3fdc0d803b8457c1f271fdcaed616b1291 Mon Sep 17 00:00:00 2001
From: Francisco Massa <fvsmassa@gmail.com>
Date: Mon, 22 Feb 2021 10:39:30 +0100
Subject: [PATCH] Revert "Added "-fopenmp" cflags (#2783) (#3006)" (#3038)

This reverts commit 74de51d6d478e289135d9274e6af550a9bfba137.
---
 setup.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/setup.py b/setup.py
index 0317d6e6483..fd653fea0f8 100644
--- 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: