Update build extensions per openmp

Author: giuliabaldini

build_extension.py

@@ -19,8 +19,6 @@
         "kmsr/cpp/yildirim.cpp",
     ],
     include_dirs=["kmsr"],
-    extra_compile_args=["-fopenmp"],
-    extra_link_args=["-lgomp"],
 )
 
 # Thank you https://github.com/dstein64/kmeans1d!
@@ -34,6 +32,8 @@ def build_extensions(self) -> None:
         # with clang++. This works across compilers (ignored by MSVC).
         for extension in self.extensions:
             extension.extra_compile_args.append("-std=c++11")
+            extension.extra_compile_args.append("-fopenmp")
+
         try:
             build_ext.build_extensions(self)
         except CompileError:
@@ -43,6 +43,7 @@ def build_extensions(self) -> None:
             for extension in self.extensions:
                 extension.extra_compile_args.append("-stdlib=libc++")
                 extension.extra_link_args.append("-stdlib=libc++")
+                extension.extra_link_args.append("-lomp")
             build_ext.build_extensions(self)