[SYCL] Testing for kernel info queries

Signed-off-by: Steffen Larsen <[email protected]>

Author: Steffen Larsen

sycl/test/basic_tests/kernel_info.cpp

@@ -0,0 +1,66 @@
+// RUN: %clangxx -fsycl -fsycl-targets=%sycl_triple %s -o %t.out -L %opencl_libs_dir -lOpenCL
+// RUN: %CPU_RUN_PLACEHOLDER %t.out
+// RUN: %GPU_RUN_PLACEHOLDER %t.out
+// RUN: %ACC_RUN_PLACEHOLDER %t.out
+
+//==--- kernel_info.cpp - SYCL kernel info test ----------------------------==//
+//
+// 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/sycl.hpp>
+
+using namespace cl::sycl;
+
+void check(bool condition, const char *conditionString, const char *filename,
+           const long line) noexcept {
+  if (!condition) {
+    std::cerr << "CHECK failed in " << filename << "#" << line << " "
+              << conditionString << "\n";
+    std::abort();
+  }
+}
+
+#define CHECK(CONDITION) check(CONDITION, #CONDITION, __FILE__, __LINE__)
+
+int main() {
+  queue q;
+
+  buffer<int, 1> buf(range<1>(1));
+  program prg(q.get_context());
+
+  prg.build_with_kernel_type<class SingleTask>();
+  kernel krn = prg.get_kernel<class SingleTask>();
+
+  q.submit([&](handler &cgh) {
+    auto acc = buf.get_access<access::mode::read_write>(cgh);
+    cgh.single_task<class SingleTask>(krn, [=]() { acc[0] = acc[0] + 1; });
+  });
+
+  const string_class krnName = krn.get_info<info::kernel::function_name>();
+  CHECK(!krnName.empty());
+  const cl_uint krnArgCount = krn.get_info<info::kernel::num_args>();
+  CHECK(krnArgCount > 0);
+  const context krnCtx = krn.get_info<info::kernel::context>();
+  CHECK(krnCtx == q.get_context());
+  const program krnPrg = krn.get_info<info::kernel::program>();
+  CHECK(krnPrg == prg);
+  const cl_uint krnRefCount = krn.get_info<info::kernel::reference_count>();
+  CHECK(krnRefCount > 0);
+  const string_class krnAttr = krn.get_info<info::kernel::attributes>();
+  CHECK(krnAttr.empty());
+
+  device dev = q.get_device();
+  const size_t wgSize =
+      krn.get_work_group_info<info::kernel_work_group::work_group_size>(dev);
+  CHECK(wgSize > 0);
+  const size_t prefWGSizeMult = krn.get_work_group_info<
+      info::kernel_work_group::preferred_work_group_size_multiple>(dev);
+  CHECK(prefWGSizeMult > 0);
+  const cl_ulong prvMemSize =
+      krn.get_work_group_info<info::kernel_work_group::private_mem_size>(dev);
+  CHECK(prvMemSize == 0);
+}