From b21ba586653b741df302c390e38a13895c7a4f7d Mon Sep 17 00:00:00 2001
From: Omar Ahmed <omar.ahmed@codeplay.com>
Date: Mon, 16 Sep 2024 10:20:10 +0100
Subject: [PATCH] Merge pull request #2060 from
 martygrant/martin/supressSystemErrorsOnWindows

[Loader] Suppress system errors when loading adapters on Windows
---
 source/loader/ur_loader.cpp | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/source/loader/ur_loader.cpp b/source/loader/ur_loader.cpp
index bfc9da3e50..7609747622 100644
--- a/source/loader/ur_loader.cpp
+++ b/source/loader/ur_loader.cpp
@@ -15,6 +15,27 @@ context_t *getContext() { return context_t::get_direct(); }
 
 ///////////////////////////////////////////////////////////////////////////////
 ur_result_t context_t::init() {
+#ifdef _WIN32
+    // Suppress system errors.
+    // Tells the system to not display the critical-error-handler message box.
+    // Instead, the system sends the error to the calling process.
+    // This is crucial for graceful handling of adapters that couldn't be
+    // loaded, e.g. due to missing native run-times.
+    // TODO: add reporting in case of an error.
+    // NOTE: we restore the old mode to not affect user app behavior.
+    // See https://github.com/intel/llvm/blob/sycl/sycl/ur_win_proxy_loader/ur_win_proxy_loader.cpp (preloadLibraries())
+    UINT SavedMode = SetErrorMode(SEM_FAILCRITICALERRORS);
+#endif
+
+#ifdef UR_STATIC_ADAPTER_LEVEL_ZERO
+    // If the adapters were force loaded, it means the user wants to use
+    // a specific adapter library. Don't load any static adapters.
+    if (!adapter_registry.adaptersForceLoaded()) {
+        auto &level_zero = platforms.emplace_back(nullptr);
+        ur::level_zero::urAdapterGetDdiTables(&level_zero.dditable.ur);
+    }
+#endif
+
     for (const auto &adapterPaths : adapter_registry) {
         for (const auto &path : adapterPaths) {
             auto handle = LibLoader::loadAdapterLibrary(path.string().c_str());
@@ -24,6 +45,10 @@ ur_result_t context_t::init() {
             }
         }
     }
+#ifdef _WIN32
+    // Restore system error handling.
+    (void)SetErrorMode(SavedMode);
+#endif
 
     forceIntercept = getenv_tobool("UR_ENABLE_LOADER_INTERCEPT");