Enable the registry of level_zero_v2

oneapi-src · Jan 29, 2025 · 58ccaa9 · 58ccaa9
1 parent 4643d1c
commit 58ccaa9
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/scripts/core/INTRO.rst b/scripts/core/INTRO.rst
@@ -404,6 +404,10 @@ Specific environment variables can be set to control the behavior of unified run
 
     This environment variable is default enabled on Linux, but default disabled on Windows.
 
+.. envvar:: UR_ADAPTER_LEVEL_ZERO_V2
+
+    If set, UR will use level_zero_v2 instead of the default level_zero.
+
 CTS Environment Variables
 -------------------------
 

diff --git a/source/loader/ur_adapter_registry.hpp b/source/loader/ur_adapter_registry.hpp
@@ -121,7 +121,7 @@ class AdapterRegistry {
   // to load the adapter.
   std::vector<std::vector<fs::path>> adaptersLoadPaths;
 
-  static constexpr std::array<const char *, 5> knownAdapterNames{
+  std::vector<const char *> knownAdapterNames{
       MAKE_LIBRARY_NAME("ur_adapter_level_zero", "0"),
       MAKE_LIBRARY_NAME("ur_adapter_opencl", "0"),
       MAKE_LIBRARY_NAME("ur_adapter_cuda", "0"),
@@ -263,6 +263,15 @@ class AdapterRegistry {
 #else
     bool loaderPreFilter = getenv_tobool("UR_LOADER_PRELOAD_FILTER", true);
 #endif
+
+    if (getenv_tobool("UR_ADAPTER_LEVEL_ZERO_V2")) {
+      knownAdapterNames.push_back(
+          MAKE_LIBRARY_NAME("ur_adapter_level_zero_v2", "0"));
+    } else {
+      knownAdapterNames.push_back(
+          MAKE_LIBRARY_NAME("ur_adapter_level_zero", "0"));
+    }
+
     for (const auto &adapterName : knownAdapterNames) {
 
       if (loaderPreFilter) {