[SYCL] Test queue::ext_oneapi_empty() API

Author: againull

SYCL/Basic/in_order_queue_status.cpp

@@ -0,0 +1,84 @@
+// RUN: %clangxx -fsycl -fsycl-targets=%sycl_triple %s -o %t.out
+// RUN: %ACC_RUN_PLACEHOLDER %t.out
+// RUN: %CPU_RUN_PLACEHOLDER %t.out
+// RUN: %GPU_RUN_PLACEHOLDER %t.out
+
+// Test checks that queue::ext_oneapi_empty() returns status of the in-order
+// queue.
+
+#include <chrono>
+#include <iostream>
+#include <sycl.hpp>
+#include <thread>
+
+static void CheckArray(int *x, size_t buffer_size, int expected) {
+  for (size_t i = 0; i < buffer_size; ++i) {
+    assert(x[i] == expected);
+  }
+}
+
+using namespace sycl;
+
+void TestFunc(queue &Q) {
+  static constexpr int Size = 100;
+
+  assert(Q.ext_oneapi_empty() && "Queue is expected to be empty");
+
+  int *X = malloc_host<int>(Size, Q);
+  int *Y = malloc_host<int>(Size, Q);
+
+  auto FillEv = Q.fill(X, 99, Size);
+  auto SingleTaskEv = Q.submit([&](handler &CGH) {
+    auto SingleTask = [=] {
+      for (int I = 0; I < Size; I++)
+        X[I] += 1;
+    };
+    CGH.single_task(SingleTask);
+  });
+  auto MemCpyEv = Q.copy(X, Y, Size);
+  constexpr int NumIter = 5;
+  for (int I = 0; I < NumIter; I++) {
+    Q.submit([&](handler &CGH) {
+      CGH.parallel_for<class Kernel1>(sycl::range<1>(Size),
+                                      [=](sycl::id<1> WI) { Y[WI] *= 2; });
+    });
+  }
+
+  // Wait a bit to give a chance for tasks to complete.
+  std::this_thread::sleep_for(std::chrono::milliseconds(500));
+
+  // We expect that all submitted tasks are finished if ext_oneapi_empty is
+  // true.
+  if (Q.ext_oneapi_empty())
+    CheckArray(Y, Size, 3200);
+
+  Q.wait();
+
+  // After synchronization queue must be empty.
+  assert(Q.ext_oneapi_empty() && "Queue is expected to be empty");
+
+  free(X, Q);
+  free(Y, Q);
+}
+
+int main() {
+  // Test in-order queue.
+  queue Q1{property::queue::in_order()};
+  TestFunc(Q1);
+
+  // Test in-order queue with discard_events property.
+  sycl::property_list Props{
+      property::queue::in_order{},
+      sycl::ext::oneapi::property::queue::discard_events{}};
+  queue Q2{Props};
+  try {
+    TestFunc(Q2);
+
+  } catch (sycl::exception &E) {
+    std::cout << E.what() << std::endl;
+    // Feature is not supported for OpenCL.
+    assert(Q2.get_device().get_backend() == backend::opencl);
+  }
+
+  return 0;
+}

SYCL/Basic/out_of_order_queue_status.cpp

@@ -0,0 +1,78 @@
+// RUN: %clangxx -fsycl -fsycl-targets=%sycl_triple %s -o %t.out
+// RUN: %ACC_RUN_PLACEHOLDER %t.out
+// RUN: %CPU_RUN_PLACEHOLDER %t.out
+// RUN: %GPU_RUN_PLACEHOLDER %t.out
+
+// Test checks that queue::ext_oneapi_empty() returns status of the out-of-order
+// queue.
+
+#include <chrono>
+#include <iostream>
+#include <sycl.hpp>
+#include <thread>
+
+static void CheckArray(int *x, size_t buffer_size, int expected) {
+  for (size_t i = 0; i < buffer_size; ++i) {
+    assert(x[i] == expected);
+  }
+}
+
+using namespace sycl;
+
+void TestFunc(queue &Q) {
+  static constexpr int Size = 100;
+
+  assert(Q.ext_oneapi_empty() && "Queue is expected to be empty");
+
+  int *X = malloc_host<int>(Size, Q);
+  int *Y = malloc_host<int>(Size, Q);
+
+  auto FillEv = Q.fill(X, 99, Size);
+  auto HostEv = Q.submit([&](handler &CGH) {
+    CGH.depends_on(FillEv);
+    auto HostTask = [=] {
+      for (int I = 0; I < Size; I++)
+        X[I] += 1;
+    };
+    CGH.host_task(HostTask);
+  });
+  auto MemCpyEv = Q.copy(X, Y, Size, {HostEv});
+  constexpr int NumIter = 5;
+  for (int I = 0; I < NumIter; I++) {
+    Q.submit([&](handler &CGH) {
+      CGH.depends_on(MemCpyEv);
+      CGH.parallel_for<class Kernel1>(
+          sycl::range<1>(Size / NumIter),
+          [=](sycl::id<1> WI) { Y[WI + I * Size / NumIter] *= 2; });
+    });
+  }
+
+  // Wait a bit to give a chance for tasks to complete.
+  std::this_thread::sleep_for(std::chrono::milliseconds(500));
+
+  // We expect that all submitted tasks are finished if ext_oneapi_empty is
+  // true.
+  if (Q.ext_oneapi_empty())
+    CheckArray(Y, Size, 200);
+
+  Q.wait();
+
+  // After synchronization queue must be empty.
+  assert(Q.ext_oneapi_empty() && "Queue is expected to be empty");
+
+  free(X, Q);
+  free(Y, Q);
+}
+
+int main() {
+  queue Q;
+
+  try {
+    TestFunc(Q);
+  } catch (sycl::exception &E) {
+    std::cout << E.what() << std::endl;
+    // Feature is not supported for OpenCL.
+    assert(Q.get_device().get_backend() == backend::opencl);
+  }
+  return 0;
+}