[SYCL][E2E] make sure zeInit in test is called after UR is initialized

UR internally sets env variables to control behavior or L0 loader
(e.g. enabling validation layer). If zeInit in test is called first
the vars will have no effect.

Author: igchor

sycl/test-e2e/Adapters/level_zero/batch_barrier.cpp

@@ -21,6 +21,12 @@ void submit_kernel(queue &q) {
 }
 
 int main(int argc, char *argv[]) {
+  queue q;
+
+  submit_kernel(q); // starts a batch
+                    // CHECK: ---> urEnqueueKernelLaunch
+                    // CHECK-NOT: zeCommandQueueExecuteCommandLists
+
   // Initialize Level Zero driver is required if this test is linked
   // statically with Level Zero loader, the driver will not be init otherwise.
   ze_result_t result = zeInit(ZE_INIT_FLAG_GPU_ONLY);
@@ -29,12 +35,6 @@ int main(int argc, char *argv[]) {
     return 1;
   }
 
-  queue q;
-
-  submit_kernel(q); // starts a batch
-                    // CHECK: ---> urEnqueueKernelLaunch
-                    // CHECK-NOT: zeCommandQueueExecuteCommandLists
-
   // continue the batch
   event barrier = q.ext_oneapi_submit_barrier();
   // CHECK: ---> urEnqueueEventsWaitWithBarrierExt

sycl/test-e2e/Adapters/level_zero/interop-buffer-multi-dim.cpp

@@ -16,14 +16,6 @@ using namespace sycl;
 int main() {
 #ifdef SYCL_EXT_ONEAPI_BACKEND_LEVEL_ZERO
   try {
-    // Initialize Level Zero driver is required if this test is linked
-    // statically with Level Zero loader, the driver will not be init otherwise.
-    ze_result_t result = zeInit(ZE_INIT_FLAG_GPU_ONLY);
-    if (result != ZE_RESULT_SUCCESS) {
-      std::cout << "zeInit failed\n";
-      return 1;
-    }
-
     platform Plt{gpu_selector_v};
 
     auto Devices = Plt.get_devices();
@@ -33,6 +25,14 @@ int main() {
       return 0;
     }
 
+    // Initialize Level Zero driver is required if this test is linked
+    // statically with Level Zero loader, the driver will not be init otherwise.
+    ze_result_t result = zeInit(ZE_INIT_FLAG_GPU_ONLY);
+    if (result != ZE_RESULT_SUCCESS) {
+      std::cout << "zeInit failed\n";
+      return 1;
+    }
+
     device Device = Devices[0];
     context Context{Device};
     queue Queue{Context, Device};

sycl/test-e2e/Adapters/level_zero/interop-buffer-ownership.cpp

@@ -118,14 +118,6 @@ void test_copyback_and_free(
 int main() {
 #ifdef SYCL_EXT_ONEAPI_BACKEND_LEVEL_ZERO
   try {
-    // Initialize Level Zero driver is required if this test is linked
-    // statically with Level Zero loader, the driver will not be init otherwise.
-    ze_result_t result = zeInit(ZE_INIT_FLAG_GPU_ONLY);
-    if (result != ZE_RESULT_SUCCESS) {
-      std::cout << "zeInit failed\n";
-      return 1;
-    }
-
     platform Plt{gpu_selector_v};
 
     auto Devices = Plt.get_devices();
@@ -135,6 +127,14 @@ int main() {
       return 0;
     }
 
+    // Initialize Level Zero driver is required if this test is linked
+    // statically with Level Zero loader, the driver will not be init otherwise.
+    ze_result_t result = zeInit(ZE_INIT_FLAG_GPU_ONLY);
+    if (result != ZE_RESULT_SUCCESS) {
+      std::cout << "zeInit failed\n";
+      return 1;
+    }
+
     device Dev1 = Devices[0];
     context Context1{Dev1};
     queue Queue1{Context1, Dev1};

sycl/test-e2e/Adapters/level_zero/interop-buffer.cpp

@@ -36,14 +36,6 @@ class DiscreteSelector : public sycl::device_selector {
 int main() {
 #ifdef SYCL_EXT_ONEAPI_BACKEND_LEVEL_ZERO
   try {
-    // Initialize Level Zero driver is required if this test is linked
-    // statically with Level Zero loader, the driver will not be init otherwise.
-    ze_result_t result = zeInit(ZE_INIT_FLAG_GPU_ONLY);
-    if (result != ZE_RESULT_SUCCESS) {
-      std::cout << "zeInit failed\n";
-      return 1;
-    }
-
     queue Queue{};
 
     auto Context = Queue.get_info<info::queue::context>();
@@ -53,6 +45,14 @@ int main() {
     auto ZeContext = get_native<backend::ext_oneapi_level_zero>(Context);
     auto ZeDevice = get_native<backend::ext_oneapi_level_zero>(Device);
 
+    // Initialize Level Zero driver is required if this test is linked
+    // statically with Level Zero loader, the driver will not be init otherwise.
+    ze_result_t result = zeInit(ZE_INIT_FLAG_GPU_ONLY);
+    if (result != ZE_RESULT_SUCCESS) {
+      std::cout << "zeInit failed\n";
+      return 1;
+    }
+
     ze_host_mem_alloc_desc_t HostDesc = {};
     HostDesc.stype = ZE_STRUCTURE_TYPE_HOST_MEM_ALLOC_DESC;
     HostDesc.pNext = nullptr;

sycl/test-e2e/Adapters/level_zero/interop-get-native-mem.cpp

@@ -23,14 +23,6 @@ constexpr size_t SIZE = 16;
 int main() {
 #ifdef SYCL_EXT_ONEAPI_BACKEND_LEVEL_ZERO
   try {
-    // Initialize Level Zero driver is required if this test is linked
-    // statically with Level Zero loader, the driver will not be init otherwise.
-    ze_result_t result = zeInit(ZE_INIT_FLAG_GPU_ONLY);
-    if (result != ZE_RESULT_SUCCESS) {
-      std::cout << "zeInit failed\n";
-      return 1;
-    }
-
     platform Plt{gpu_selector_v};
 
     auto Devices = Plt.get_devices();
@@ -40,6 +32,14 @@ int main() {
       return 0;
     }
 
+    // Initialize Level Zero driver is required if this test is linked
+    // statically with Level Zero loader, the driver will not be init otherwise.
+    ze_result_t result = zeInit(ZE_INIT_FLAG_GPU_ONLY);
+    if (result != ZE_RESULT_SUCCESS) {
+      std::cout << "zeInit failed\n";
+      return 1;
+    }
+
     device Dev1 = Devices[0];
     context Context1{Dev1};
     queue Queue1{Context1, Dev1};

sycl/test-e2e/Adapters/level_zero/interop-image-get-native-mem.cpp

@@ -42,6 +42,10 @@ using namespace sycl;
 
 int main() {
 #ifdef SYCL_EXT_ONEAPI_BACKEND_LEVEL_ZERO
+  constexpr auto BE = sycl::backend::ext_oneapi_level_zero;
+  sycl::device D =
+      sycl::ext::oneapi::filter_selector("level_zero:gpu").select_device();
+
   // Initialize Level Zero driver is required if this test is linked
   // statically with Level Zero loader, the driver will not be init otherwise.
   ze_result_t result = zeInit(ZE_INIT_FLAG_GPU_ONLY);
@@ -50,10 +54,6 @@ int main() {
     return 1;
   }
 
-  constexpr auto BE = sycl::backend::ext_oneapi_level_zero;
-  sycl::device D =
-      sycl::ext::oneapi::filter_selector("level_zero:gpu").select_device();
-
   sycl::context Ctx{D};
   sycl::queue Q(Ctx, D);
   auto ZeContext = sycl::get_native<BE>(Ctx);

sycl/test-e2e/Adapters/level_zero/interop-image-ownership.cpp

@@ -144,6 +144,8 @@ void test(sycl::ext::oneapi::level_zero::ownership Ownership) {
 
 int main() {
 #ifdef SYCL_EXT_ONEAPI_BACKEND_LEVEL_ZERO
+  platform Plt{gpu_selector_v};
+
   // Initialize Level Zero driver is required if this test is linked
   // statically with Level Zero loader, the driver will not be init otherwise.
   ze_result_t result = zeInit(ZE_INIT_FLAG_GPU_ONLY);

sycl/test-e2e/Adapters/level_zero/interop-image.cpp

@@ -32,6 +32,12 @@ using namespace sycl;
 
 int main() {
 #ifdef SYCL_EXT_ONEAPI_BACKEND_LEVEL_ZERO
+  constexpr auto BE = sycl::backend::ext_oneapi_level_zero;
+
+  platform Plt{gpu_selector_v};
+
+  auto Devices = Plt.get_devices();
+
   // Initialize Level Zero driver is required if this test is linked
   // statically with Level Zero loader, the driver will not be init otherwise.
   ze_result_t result = zeInit(ZE_INIT_FLAG_GPU_ONLY);
@@ -40,12 +46,6 @@ int main() {
     return 1;
   }
 
-  constexpr auto BE = sycl::backend::ext_oneapi_level_zero;
-
-  platform Plt{gpu_selector_v};
-
-  auto Devices = Plt.get_devices();
-
   if (Devices.size() < 1) {
     std::cout << "Devices not found" << std::endl;
     return 0;

sycl/test-e2e/Adapters/level_zero/interop-keep-ownership.cpp

@@ -14,6 +14,10 @@
 using namespace sycl;
 
 int main() {
+  // Creat SYCL platform/device
+  device Device(gpu_selector_v);
+  platform Platform = Device.get_info<info::device::platform>();
+
   // Initialize Level Zero driver is required if this test is linked
   // statically with Level Zero loader, the driver will not be init otherwise.
   ze_result_t result = zeInit(ZE_INIT_FLAG_GPU_ONLY);
@@ -22,10 +26,6 @@ int main() {
     return 1;
   }
 
-  // Creat SYCL platform/device
-  device Device(gpu_selector_v);
-  platform Platform = Device.get_info<info::device::platform>();
-
   // Create native Level-Zero context
   ze_context_handle_t ZeContext;
   ze_context_desc_t ZeContextDesc = {ZE_STRUCTURE_TYPE_CONTEXT_DESC, nullptr,

sycl/test-e2e/Graph/Inputs/interop-level-zero-launch-kernel.cpp

@@ -37,6 +37,12 @@ std::vector<uint8_t> loadSpirvFromFile(std::string FileName) {
 }
 
 int main(int, char **argv) {
+  device Device;
+  if (!getDevice(Device, backend::ext_oneapi_level_zero)) {
+    // No suitable device found.
+    return 0;
+  }
+
   // Initialize Level Zero driver is required if this test is linked
   // statically with Level Zero loader, the driver will not be init otherwise.
   ze_result_t result = zeInit(ZE_INIT_FLAG_GPU_ONLY);
@@ -45,12 +51,6 @@ int main(int, char **argv) {
     return 1;
   }
 
-  device Device;
-  if (!getDevice(Device, backend::ext_oneapi_level_zero)) {
-    // No suitable device found.
-    return 0;
-  }
-
   std::vector<uint8_t> Spirv = loadSpirvFromFile(argv[1]);
 
   const sycl::context Context{Device};