add omp dirs for macos

Author: lrntct

setup.py

@@ -20,8 +20,14 @@
     "/opt/homebrew/include",
     "/usr/local/include",
     "/opt/homebrew/opt/llvm/include",
+    "/opt/homebrew/opt/libomp/lib",
+]
+macos_dirs = [
+    "/opt/homebrew/lib",
+    "/usr/local/lib",
+    "/opt/homebrew/opt/llvm/include",
+    "/opt/homebrew/opt/libomp/include",
 ]
-macos_dirs = ["/opt/homebrew/lib", "/usr/local/lib", "/opt/homebrew/opt/llvm/include"]
 
 
 class build_ext_compiler_check(build_ext):