test to expose zeVirtualMemQueryPageSize not using allocation size

Author: lslusarczyk

sycl/test-e2e/Graph/AsyncAlloc/Explicit/async_alloc_different_sizes.cpp

@@ -0,0 +1,10 @@
+// RUN: %{build} -o %t.out
+// RUN: %{run} %t.out
+// 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 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 2>&1 | FileCheck %s --implicit-check-not=LEAK %}
+
+#define GRAPH_E2E_EXPLICIT
+
+#include "../Inputs/async_alloc_different_sizes.cpp"

sycl/test-e2e/Graph/AsyncAlloc/Inputs/async_alloc_different_sizes.cpp

@@ -0,0 +1,34 @@
+// Tests queries associated with graph-owned allocations
+
+#include "../../graph_common.hpp"
+#include <sycl/ext/oneapi/experimental/async_alloc/async_alloc.hpp>
+
+void asyncAllocWorksWithSize(size_t Size) {
+	queue Queue{};
+
+	exp_ext::command_graph Graph{Queue.get_context(), Queue.get_device()};
+
+	void *AsyncPtr = nullptr;
+	// Add alloc node
+	auto AllocNode = add_node(Graph, Queue, [&](handler &CGH) {
+	  AsyncPtr = exp_ext::async_malloc(CGH, usm::alloc::device, Size);
+	});
+
+	add_node(
+		Graph, Queue,
+		[&](handler &CGH) {
+		  depends_on_helper(CGH, AllocNode);
+		  exp_ext::async_free(CGH, AsyncPtr);
+		},
+		AllocNode);
+
+	auto GraphExec = Graph.finalize();
+}
+
+int main() {
+	asyncAllocWorksWithSize(1);
+	asyncAllocWorksWithSize(131);
+	asyncAllocWorksWithSize(10071);
+	asyncAllocWorksWithSize(1007177);
+	//asyncAllocWorksWithSize(191439360); // BUG
+}

sycl/test-e2e/Graph/AsyncAlloc/RecordReplay/async_alloc_different_sizes.cpp

@@ -0,0 +1,10 @@
+// RUN: %{build} -o %t.out
+// RUN: %{run} %t.out
+// 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 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 2>&1 | FileCheck %s --implicit-check-not=LEAK %}
+
+#define GRAPH_E2E_RECORD_REPLAY
+
+#include "../Inputs/async_alloc_queries.cpp"