[SYCL] E2E test for interop_task with Level-Zero and OpenCL (#252)

Author: mikhail-nikolskiy

SYCL/Plugin/interop-level-zero-interop-task-mem.cpp

@@ -2,45 +2,54 @@
 // RUN: %clangxx -fsycl -fsycl-targets=%sycl_triple %level_zero_options %s -o %t.out
 // RUN: env SYCL_BE=PI_LEVEL_ZERO %GPU_RUN_PLACEHOLDER %t.out
 
-// Test fails on Level Zero on Linux
-// UNSUPPORTED: level_zero && linux
+// Test for Level Zero interop_task.
 
-// Test for Level Zero interop_task
+// Level-Zero
+#include <level_zero/ze_api.h>
 
+// SYCL
 #include <CL/sycl.hpp>
-// clang-format off
-#include <level_zero/ze_api.h>
 #include <CL/sycl/backend/level_zero.hpp>
-// clang-format on
-
-class my_selector : public cl::sycl::device_selector {
-public:
-  int operator()(const cl::sycl::device &dev) const override {
-    return (dev.get_platform().get_backend() == cl::sycl::backend::level_zero)
-               ? 1
-               : 0;
-  }
-};
+
+using namespace sycl;
+
+constexpr size_t SIZE = 16;
 
 int main() {
-  sycl::queue sycl_queue = sycl::queue(my_selector());
-
-  ze_context_handle_t ze_context =
-      sycl_queue.get_context().get_native<sycl::backend::level_zero>();
-  std::cout << "zeContextGetStatus = " << zeContextGetStatus(ze_context)
-            << std::endl;
-
-  auto buf = cl::sycl::buffer<uint8_t, 1>(1024);
-  sycl_queue.submit([&](cl::sycl::handler &cgh) {
-    auto acc = buf.get_access<cl::sycl::access::mode::write>(cgh);
-    cgh.interop_task([&](const cl::sycl::interop_handler &ih) {
-      void *device_ptr = ih.get_mem<sycl::backend::level_zero>(acc);
-      ze_memory_allocation_properties_t memAllocProperties{};
-      zeMemGetAllocProperties(ze_context, device_ptr, &memAllocProperties,
-                              nullptr);
-      std::cout << "Memory type = " << memAllocProperties.type << std::endl;
-    });
-  });
+  queue queue{};
+
+  try {
+    buffer<uint8_t, 1> buffer(SIZE);
+    image<2> image(image_channel_order::rgba, image_channel_type::fp32,
+                   {SIZE, SIZE});
+
+    ze_context_handle_t ze_context =
+        queue.get_context().get_native<backend::level_zero>();
+
+    queue
+        .submit([&](handler &cgh) {
+          auto buffer_acc = buffer.get_access<access::mode::write>(cgh);
+          auto image_acc = image.get_access<float4, access::mode::write>(cgh);
+          cgh.interop_task([=](const interop_handler &ih) {
+            void *device_ptr = ih.get_mem<backend::level_zero>(buffer_acc);
+            ze_memory_allocation_properties_t memAllocProperties{};
+            ze_result_t res = zeMemGetAllocProperties(
+                ze_context, device_ptr, &memAllocProperties, nullptr);
+            assert(res == ZE_RESULT_SUCCESS);
+
+            ze_image_handle_t ze_image =
+                ih.get_mem<backend::level_zero>(image_acc);
+            assert(ze_image != nullptr);
+          });
+        })
+        .wait();
+  } catch (exception const &e) {
+    std::cout << "SYCL exception caught: " << e.what() << std::endl;
+    return e.get_cl_code();
+  } catch (const char *msg) {
+    std::cout << "Exception caught: " << msg << std::endl;
+    return 1;
+  }
 
   return 0;
 }

SYCL/Plugin/interop-opencl-interop-task-mem.cpp

@@ -0,0 +1,52 @@
+// REQUIRES: opencl, opencl_icd
+
+// RUN: %clangxx -fsycl -fsycl-targets=%sycl_triple %s -o %t.out %opencl_lib
+// RUN: %GPU_RUN_PLACEHOLDER %t.out
+
+// Test for OpenCL interop_task.
+
+#include <CL/opencl.h>
+#include <CL/sycl.hpp>
+#include <CL/sycl/backend/opencl.hpp>
+
+using namespace sycl;
+
+constexpr size_t SIZE = 16;
+
+int main() {
+  queue queue{};
+
+  try {
+    buffer<uint8_t, 1> buffer(SIZE);
+    image<2> image(image_channel_order::rgba, image_channel_type::fp32,
+                   {SIZE, SIZE});
+
+    queue
+        .submit([&](handler &cgh) {
+          auto buffer_acc = buffer.get_access<access::mode::write>(cgh);
+          auto image_acc = image.get_access<float4, access::mode::write>(cgh);
+          cgh.interop_task([=](const interop_handler &ih) {
+            cl_mem buffer_mem = ih.get_mem<backend::opencl>(buffer_acc);
+            size_t size = 0;
+            clGetMemObjectInfo(buffer_mem, CL_MEM_SIZE, sizeof(size),
+                               (void *)&size, nullptr);
+            assert(size == SIZE);
+
+            cl_mem mem = ih.get_mem<backend::opencl>(image_acc);
+            size_t width = 0;
+            clGetImageInfo(mem, CL_IMAGE_WIDTH, sizeof(width), (void *)&width,
+                           nullptr);
+            assert(width == SIZE);
+          });
+        })
+        .wait();
+  } catch (exception const &e) {
+    std::cout << "SYCL exception caught: " << e.what() << std::endl;
+    return e.get_cl_code();
+  } catch (const char *msg) {
+    std::cout << "Exception caught: " << msg << std::endl;
+    return 1;
+  }
+
+  return 0;
+}