Try new way to find omp library

Author: mcbarton

test/test_xcpp_kernel.py

@@ -134,17 +134,19 @@ class XCppTests(jupyter_kernel_test.KernelTests):
         code_omp="""
     #include <omp.h>
     #include <iostream>
-    #include <clang/Interpreter/CppInterOp.h>
-    bool loaded=Cpp::LoadLibrary("libomp");
-    if(loaded) {
-      int max_threads = omp_get_max_threads();
-      omp_set_thread_num(max_threads);
-      std::cerr<<"OpenMP library loaded"<<std::endl;
-    }
-    else 
-    {
-        std::cerr<<"OpenMP library not loaded"<<std::endl;
-    }
+    #include <clang/Interpreter/CppInterOp.h>  
+    #include "llvm/Support/FileSystem.h"
+    #include "llvm/Support/Path.h"
+    std::string BinaryPath = GetExecutablePath(/*Argv0=*/nullptr);
+    llvm::StringRef Dir = llvm::sys::path::parent_path(BinaryPath);
+    Cpp::AddSearchPath(Dir.str().c_str());
+    #ifdef __APPLE__
+      std::string PathToTestSharedLib = Cpp::SearchLibrariesForSymbol("_omp_get_max_threads_", /*system_search=*/false);
+    #else
+      std::string PathToTestSharedLib = Cpp::SearchLibrariesForSymbol("omp_get_max_threads_", /*system_search=*/false);
+    #endif // __APPLE__
+    bool loaded=Cpp::LoadLibrary("PathToTestSharedLib.c_str())");
+    std::cerr<<loaded<<std::endl;
     """
         def test_xcpp_omp(self):
             self.flush_channels()