fix a bug in setup.py

tanshen · Jun 25, 2016 · 35e7a86 · 35e7a86
1 parent 99fa211
commit 35e7a86
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fast-rcnn/lib/setup.py b/fast-rcnn/lib/setup.py
@@ -108,12 +108,12 @@ def build_extensions(self):
     Extension(
         "utils.cython_bbox",
         ["utils/bbox.pyx"],
-        extra_compile_args=["-Wno-cpp", "-Wno-unused-function"],
+        extra_compile_args={'gcc': ["-Wno-cpp", "-Wno-unused-function"]},
     ),
     Extension(
         "utils.cython_nms",
         ["utils/nms.pyx"],
-        extra_compile_args=["-Wno-cpp", "-Wno-unused-function"],
+        extra_compile_args={'gcc': ["-Wno-cpp", "-Wno-unused-function"]},
     ),
     Extension(
         "nms.cpu_nms",