Add preliminary implementation of tests via syclos side

Author: ianayl

sycl/cmake/modules/FetchUnifiedRuntime.cmake

@@ -121,7 +121,7 @@ if(SYCL_UR_USE_FETCH_CONTENT)
   # Author: Maosu Zhao <maosu.zhao@intel.com>
   # Date:   Thu Oct 17 16:31:21 2024 +0800
   #     [DeviceSanitizer] Refactor the code to manage shadow memory (#2127)
-  set(UNIFIED_RUNTIME_TAG 644eef9f8dfcc3931fd2883f24eace60ca98b3ed)
+  set(UNIFIED_RUNTIME_TAG ca43f937d91c53e04ebad69a02dbffdb1bc2f78e)
 
   set(UMF_BUILD_EXAMPLES OFF CACHE INTERNAL "EXAMPLES")
   # Due to the use of dependentloadflag and no installer for UMF and hwloc we need

sycl/test-e2e/USM/prefetch_exp.cpp

@@ -30,8 +30,8 @@ int main() {
     for (int i = 0; i < count; i++)
       src[i] = i;
 
-    // Test host to device prefetch_exp(handler &CGH, ..)
     {
+      // Test host to device handler::ext_oneapi_prefetch_exp
       event init_prefetch = q.submit(
           [&](handler &cgh) { cgh.ext_oneapi_prefetch_exp(src, sizeof(float) * count); });
 
@@ -41,12 +41,27 @@ int main() {
           for (int i = 0; i < count; i++)
             dest[i] = 2 * src[i];
         });
-      }
+      });
       q.wait_and_throw();
 
       for (int i = 0; i < count; i++) {
         assert(dest[i] == i * 2);
       }
+
+      // Test device to host handler::ext_oneapi_prefetch_exp
+      q.submit([&](handler &cgh) {
+        cgh.single_task<class quadruple_dest>([=]() {
+          for (int i = 0; i < count; i++)
+            dest[i] = 4 * src[i];
+        });
+      });
+      event init_prefetch_back = q.submit(
+          [&](handler &cgh) { cgh.ext_oneapi_prefetch_exp(dest, sizeof(float) * count, sycl::ext::oneapi::experimental::migration_direction::DEVICE_TO_HOST); });
+      q.wait_and_throw();
+
+      for (int i = 0; i < count; i++) {
+        assert(dest[i] == i * 4);
+      }
     }
 
     // Test queue::prefetch
@@ -55,7 +70,7 @@ int main() {
 
       q.submit([&](handler &cgh) {
         cgh.depends_on(init_prefetch);
-        cgh.single_task<class double_dest3>([=]() {
+        cgh.single_task<class triple_dest>([=]() {
           for (int i = 0; i < count; i++)
             dest[i] = 3 * src[i];
         });
@@ -65,6 +80,19 @@ int main() {
       for (int i = 0; i < count; i++) {
         assert(dest[i] == i * 3);
       }
+
+      q.submit([&](handler &cgh) {
+        cgh.single_task<class sixtuple_dest>([=]() {
+          for (int i = 0; i < count; i++)
+            dest[i] = 6 * src[i];
+        });
+      });
+      event init_prefetch_back = q.ext_oneapi_prefetch_exp(src, sizeof(float) * count, sycl::ext::oneapi::experimental::migration_direction::DEVICE_TO_HOST);
+      q.wait_and_throw();
+
+      for (int i = 0; i < count; i++) {
+        assert(dest[i] == i * 6);
+      }
     }
     free(src, q);
     free(dest, q);

sycl/unittests/Extensions/CMakeLists.txt

@@ -16,6 +16,7 @@ add_sycl_unittest(ExtensionsTests OBJECT
   KernelProperties.cpp
   NoDeviceIPVersion.cpp
   GetLastEvent.cpp
+  USMPrefetchExp.cpp
 )
 
 add_subdirectory(CommandGraph)

sycl/unittests/Extensions/USMPrefetchExp.cpp

@@ -0,0 +1,55 @@
+#include <helpers/UrMock.hpp>
+#include <sycl/detail/core.hpp>
+#include <sycl/usm.hpp>
+#include <sycl/ext/oneapi/experimental/USM/prefetch_exp.hpp>
+
+#include <gtest/gtest.h>
+
+static constexpr int N = 8;
+static ur_usm_migration_flags_t urUSMPrefetchDirection = -1;
+
+// TODO: FIGURE OUT WHEN COMMANDBUF GETS CALLED AND IMPLEMENT COMMANDBUF TESTING
+
+ur_result_t redefinedEnqueueUSMPrefetch(void *pParams) {
+  auto params = *static_cast<ur_enqueue_usm_prefetch_params_t *>(pParams);
+  urUSMPrefetchDirection = *(params.pflags);
+  return UR_RESULT_SUCCESS;
+}
+
+TEST(USMPrefetchExp, CheckURCall) {
+  using namespace sycl;
+  unittest::UrMock<> Mock;
+  mock::getCallbacks().set_replace_callback("urEnqueueUSMPrefetch",
+                                            &redefinedEnqueueUSMPrefetch);
+  queue q;
+  int *mem = (int*) malloc_shared(sizeof(int) * N, q.get_device(), q.get_context());
+
+  // Check handler calls:
+  q.submit([&](handler &cgh) { cgh.ext_oneapi_prefetch_exp(mem, sizeof(int) * N); });
+  q.wait_and_throw();
+  EXPECT_EQ(urUSMPrefetchDirection, UR_USM_MIGRATION_FLAG_HOST_TO_DEVICE);
+
+  q.submit([&](handler &cgh) { cgh.ext_oneapi_prefetch_exp(mem, sizeof(int) * N, sycl::ext::oneapi::experimental::migration_direction::HOST_TO_DEVICE); });
+  q.wait_and_throw();
+  EXPECT_EQ(urUSMPrefetchDirection, UR_USM_MIGRATION_FLAG_HOST_TO_DEVICE);
+
+  q.submit([&](handler &cgh) { cgh.ext_oneapi_prefetch_exp(mem, sizeof(int) * N, sycl::ext::oneapi::experimental::migration_direction::DEVICE_TO_HOST); });
+  q.wait_and_throw();
+  EXPECT_EQ(urUSMPrefetchDirection, UR_USM_MIGRATION_FLAG_DEVICE_TO_HOST);
+
+  // Check queue calls:
+  q.ext_oneapi_prefetch_exp(mem, sizeof(int) * N);
+  q.wait_and_throw();
+  EXPECT_EQ(urUSMPrefetchDirection, UR_USM_MIGRATION_FLAG_HOST_TO_DEVICE);
+
+  q.ext_oneapi_prefetch_exp(mem, sizeof(int) * N, sycl::ext::oneapi::experimental::migration_direction::HOST_TO_DEVICE);
+  q.wait_and_throw();
+  EXPECT_EQ(urUSMPrefetchDirection, UR_USM_MIGRATION_FLAG_HOST_TO_DEVICE);
+
+  q.ext_oneapi_prefetch_exp(mem, sizeof(int) * N, sycl::ext::oneapi::experimental::migration_direction::DEVICE_TO_HOST);
+  q.wait_and_throw();
+  EXPECT_EQ(urUSMPrefetchDirection, UR_USM_MIGRATION_FLAG_DEVICE_TO_HOST);
+
+  // TODO: not sure what else to test for, check event? I don't think there's any other
+  // parameters to validate....
+}