[SYCL] Add unittest

Signed-off-by: Sergey V Maslov <[email protected]>

Author: smaslov-intel

sycl/unittests/pi/CMakeLists.txt

@@ -7,6 +7,7 @@ add_sycl_unittest(PiTests OBJECT
   PiMock.cpp
   PlatformTest.cpp
   pi_arguments_handler.cpp
+  piInteropRetain.cpp
 )
 
 add_dependencies(PiTests sycl)

sycl/unittests/pi/piInteropRetain.cpp

@@ -0,0 +1,65 @@
+//==--------------------- Wait.cpp --- queue unit tests --------------------==//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#include <CL/opencl.h>
+#include <CL/sycl.hpp>
+#include <CL/sycl/backend/opencl.hpp>
+
+#include <gtest/gtest.h>
+#include <helpers/PiMock.hpp>
+#include <detail/queue_impl.hpp>
+
+namespace {
+using namespace cl::sycl;
+
+static bool QueueRetainCalled;
+pi_result unexpectedQueueRetain(pi_queue Queue) { return PI_INVALID_QUEUE; }
+pi_result expectedQueueRetain(pi_queue Queue) { QueueRetainCalled = true; return PI_SUCCESS; }
+
+bool preparePiMock(platform &Plt) {
+  if (Plt.is_host()) {
+    std::cout << "Not run on host - no PI events created in that case"
+              << std::endl;
+    return false;
+  }
+  if (detail::getSyclObjImpl(Plt)->getPlugin().getBackend() !=
+      backend::opencl) {
+    std::cout << "Not run on non-OpenCL backend"
+              << std::endl;
+    return false;
+  }
+  return true;
+}
+
+TEST(PiInteropTest, CheckRetain) {
+  platform Plt{default_selector()};
+  if (!preparePiMock(Plt))
+    return;
+  context Ctx{Plt.get_devices()[0]};
+
+  unittest::PiMock Mock{Plt};
+
+  // The queue construction should not call to piQueueRetain. Instead piQueueCreate should return the "retained" queue.
+  QueueRetainCalled = false;
+  Mock.redefine<detail::PiApiKind::piQueueRetain>(unexpectedQueueRetain);
+  queue Q{Ctx, default_selector()};
+  EXPECT_FALSE(QueueRetainCalled);
+
+  QueueRetainCalled = false;
+  Mock.redefine<detail::PiApiKind::piQueueRetain>(expectedQueueRetain);
+  cl_command_queue OCLQ = get_native<backend::opencl>(Q);
+  EXPECT_TRUE(QueueRetainCalled);
+
+  // The make_queue should not call to piQueueRetain. The piextCreateQueueWithNative handle should do the "retain" if needed.
+  QueueRetainCalled = false;
+  Mock.redefine<detail::PiApiKind::piQueueRetain>(unexpectedQueueRetain);
+  queue Q1 = make_queue<backend::opencl>(OCLQ, Ctx);
+  EXPECT_FALSE(QueueRetainCalled);
+}
+
+} // namespace