setup.py.diff

diff --git a/setup.py b/setup.py
index b8b003b..e8bf426 100644
--- a/setup.py
+++ b/setup.py
@@ -1,9 +1,16 @@
 from setuptools import setup, Extension
 from glob import glob
+import os
+
+os.environ["CC"] = "/usr/local/bin/clang-omp -fopenmp"
+os.environ["CXX"] = "/usr/local/bin/clang-omp++ -fopenmp"
+
+OPENMP_VER = 20150401
 
 library = ('primesieve', dict(
     sources=glob("lib/primesieve/src/primesieve/*.cpp"),
-    include_dirs=["lib/primesieve/include"],
+    include_dirs=["lib/primesieve/include","/usr/local/include/libiomp"],
+#    macros=[('_OPENMP',OPENMP_VER)],
     language="c++",
     ))
 
@@ -16,8 +23,12 @@ else:
 extension = Extension(
         "primesieve",
         ["primesieve/primesieve.pyx"] if cythonize else ["primesieve/primesieve.cpp"],
-        include_dirs=["lib/primesieve/include", "lib/primesieve/include/primesieve"],
+        include_dirs=["lib/primesieve/include", "lib/primesieve/include/primesieve","/usr/local/include/libiomp"],
         language="c++",
+        compiler="/usr/local/bin/clang-omp -fopenmp",
+#        extra_compile_args=['-D_OPENMP=%d' % (OPENMP_VER)],
+        library_dirs=['/usr/local/lib'],
+        libraries=["iomp5"]
         )
 
 ext_modules = cythonize(extension) if cythonize else [extension]