Merge pull request oneapi-src#1673 from kswiecicki/adapters-search-wi…

…n-fix [UR] Fix adapter search on Windows
KseniyaTikhomirova · Jun 5, 2024 · 5136177 · 5136177
2 parents f06bc02 + 5263630
commit 5136177
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/source/loader/windows/adapter_search.cpp b/source/loader/windows/adapter_search.cpp
@@ -23,8 +23,14 @@ namespace fs = filesystem;
 namespace ur_loader {
 
 std::optional<fs::path> getLoaderLibPath() {
+    HMODULE hModule = NULL;
     char pathStr[MAX_PATH_LEN_WIN];
-    if (GetModuleFileNameA(nullptr, pathStr, MAX_PATH_LEN_WIN)) {
+
+    if (GetModuleHandleEx(GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS |
+                              GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT,
+                          reinterpret_cast<LPCSTR>(&getLoaderLibPath),
+                          &hModule) &&
+        GetModuleFileNameA(hModule, pathStr, MAX_PATH_LEN_WIN)) {
         auto libPath = fs::path(pathStr);
         if (fs::exists(libPath)) {
             return fs::absolute(libPath).parent_path();