Add CFLAGS

setup.py

@@ -12,12 +12,16 @@ import numpy
 
 os.environ["CC"] = "mpicc"
 
-cflags = ["-Werror", "-Wall", "-Wextra",
+cflags = ["-Werror",
+          "-Wall",
+          "-Wextra",
           # disables some warnings due to python
-          "-Wno-unused-parameter", "-Wno-strict-prototypes",
+          "-Wno-unused-parameter",
+          "-Wno-strict-prototypes",
           "-Wno-unused-function",
           "-Wno-incompatible-pointer-types",
-          "-Wno-missing-field-initializers"]
+          "-Wno-missing-field-initializers",
+]
 
 # deal with arguments
 def parseCmdLine(arg, store=False):
@@ -103,7 +107,8 @@ ext_modules = Extension("pyswiftsim.wrapper",
                         c_src,
                         include_dirs=include,
                         libraries=lib,
-                        library_dirs=lib_dir)
+                        library_dirs=lib_dir,
+                        extra_compile_args=cflags)
 
 ext_modules = [ext_modules]