From 8dd02092797852a2fdc5a11cbda7ea88518ecaf0 Mon Sep 17 00:00:00 2001
From: Ben Tracy <ben.tracy@codeplay.com>
Date: Thu, 24 Apr 2025 10:49:36 +0100
Subject: [PATCH] [SYCL][Graph] Add E2E test for kernel bundle regression

- Add test which was observed to be hanging after recent kernel bundle changes
- Test is modified version of update test that removes update functionality so it runs on current CI
---
 .../Explicit/regression_accessor_spv.cpp      | 58 +++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 sycl/test-e2e/Graph/Explicit/regression_accessor_spv.cpp

diff --git a/sycl/test-e2e/Graph/Explicit/regression_accessor_spv.cpp b/sycl/test-e2e/Graph/Explicit/regression_accessor_spv.cpp
new file mode 100644
index 0000000000000..d78200d21405b
--- /dev/null
+++ b/sycl/test-e2e/Graph/Explicit/regression_accessor_spv.cpp
@@ -0,0 +1,58 @@
+// RUN: %{build} -o %t.out
+// RUN: %{run} %t.out %S/../Inputs/Kernels/update_with_indices_accessor.spv
+// Extra run to check for leaks in Level Zero using UR_L0_LEAKS_DEBUG
+// RUN: %if level_zero %{env SYCL_PI_LEVEL_ZERO_USE_IMMEDIATE_COMMANDLISTS=0 %{l0_leak_check} %{run} %t.out %S/../Inputs/Kernels/update_with_indices_accessor.spv 2>&1 | FileCheck %s --implicit-check-not=LEAK %}
+// Extra run to check for immediate-command-list in Level Zero
+// RUN: %if level_zero %{env SYCL_PI_LEVEL_ZERO_USE_IMMEDIATE_COMMANDLISTS=1 %{l0_leak_check} %{run} %t.out %S/../Inputs/Kernels/update_with_indices_accessor.spv 2>&1 | FileCheck %s --implicit-check-not=LEAK %}
+
+// REQUIRES: level_zero
+
+// Modified version of Update/update_with_indices_accessor_spv.cpp that does
+// not require the full graph aspect, test was hanging after some changes to
+// kernel bundles so adding this test for the CI which doesn't support update
+
+#include "../graph_common.hpp"
+
+int main(int, char **argv) {
+  queue Queue{};
+  sycl::kernel_bundle KernelBundle = loadKernelsFromFile(Queue, argv[1]);
+  const auto getKernel =
+      [](sycl::kernel_bundle<sycl::bundle_state::executable> &bundle,
+         const std::string &name) {
+        return bundle.ext_oneapi_get_kernel(name);
+      };
+
+  kernel kernel = getKernel(
+      KernelBundle, "_ZTSZZ4mainENKUlRN4sycl3_V17handlerEE_clES2_EUlvE_");
+
+  const size_t N = 1024;
+
+  std::vector<int> HostDataA(N, 0);
+
+  buffer BufA{HostDataA};
+  BufA.set_write_back(false);
+
+  {
+    exp_ext::command_graph Graph{
+        Queue.get_context(),
+        Queue.get_device(),
+        {exp_ext::property::graph::assume_buffer_outlives_graph{}}};
+
+    Graph.add([&](handler &cgh) {
+      auto Acc = BufA.get_access(cgh);
+
+      cgh.set_arg(0, Acc);
+      cgh.single_task(kernel);
+    });
+
+    auto ExecGraph = Graph.finalize();
+    Queue.ext_oneapi_graph(ExecGraph).wait();
+  }
+  // Copy data back to host
+  Queue.copy(BufA.get_access(), HostDataA.data()).wait();
+
+  for (size_t i = 0; i < N; i++) {
+    assert(HostDataA[i] == i);
+  }
+  return 0;
+}