From 86d985a42f4e077e2bcca4f6950200cf98887ba0 Mon Sep 17 00:00:00 2001
From: Jason Volk <jason@zemos.net>
Date: Sat, 3 Apr 2021 12:23:52 -0700
Subject: [PATCH] ircd::cl: Add kernel workgroup info convenience interface.

---
 include/ircd/cl.h |  6 ++++++
 ircd/cl.cc        | 50 +++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 56 insertions(+)

diff --git a/include/ircd/cl.h b/include/ircd/cl.h
index 991bd0a9a..9cc294c40 100644
--- a/include/ircd/cl.h
+++ b/include/ircd/cl.h
@@ -111,6 +111,12 @@ struct ircd::cl::kern
 	void *handle {nullptr};
 
   public:
+	std::array<size_t, 3> compile_group_size(void *dev = nullptr) const;
+	size_t preferred_group_size_multiple(void *dev = nullptr) const;
+	size_t group_size(void *dev = nullptr) const;
+	size_t local_mem_size(void *dev = nullptr) const;
+	size_t stack_mem_size(void *dev = nullptr) const;
+
 	void arg(const int, data &);
 	void arg(const int, const const_buffer &);
 	template<class T> void arg(const int, const T &);
diff --git a/ircd/cl.cc b/ircd/cl.cc
index 57a7db317..e22baec47 100644
--- a/ircd/cl.cc
+++ b/ircd/cl.cc
@@ -939,6 +939,56 @@ ircd::cl::kern::arg(const int i,
 	call(clSetKernelArg, cl_kernel(handle), i, ircd::size(buf), ircd::data(buf));
 }
 
+std::array<size_t, 3>
+ircd::cl::kern::compile_group_size(void *const dev)
+const
+{
+	char buf[24];
+	const auto handle(cl_kernel(this->handle));
+	constexpr auto flag(CL_KERNEL_COMPILE_WORK_GROUP_SIZE);
+	return info<std::array<size_t, 3>>(clGetKernelWorkGroupInfo, handle, cl_device_id(dev), flag, buf);
+}
+
+size_t
+ircd::cl::kern::preferred_group_size_multiple(void *const dev)
+const
+{
+	char buf[16];
+	const auto handle(cl_kernel(this->handle));
+	constexpr auto flag(CL_KERNEL_PREFERRED_WORK_GROUP_SIZE_MULTIPLE);
+	return info<size_t>(clGetKernelWorkGroupInfo, handle, cl_device_id(dev), flag, buf);
+}
+
+size_t
+ircd::cl::kern::group_size(void *const dev)
+const
+{
+	char buf[16];
+	const auto handle(cl_kernel(this->handle));
+	constexpr auto flag(CL_KERNEL_WORK_GROUP_SIZE);
+	return info<size_t>(clGetKernelWorkGroupInfo, handle, cl_device_id(dev), flag, buf);
+}
+
+size_t
+ircd::cl::kern::local_mem_size(void *const dev)
+const
+{
+	char buf[16];
+	const auto handle(cl_kernel(this->handle));
+	constexpr auto flag(CL_KERNEL_LOCAL_MEM_SIZE);
+	return info<ulong>(clGetKernelWorkGroupInfo, handle, cl_device_id(dev), flag, buf);
+}
+
+size_t
+ircd::cl::kern::stack_mem_size(void *const dev)
+const
+{
+	char buf[16];
+	const auto handle(cl_kernel(this->handle));
+	constexpr auto flag(CL_KERNEL_PRIVATE_MEM_SIZE);
+	return info<ulong>(clGetKernelWorkGroupInfo, handle, cl_device_id(dev), flag, buf);
+}
+
 //
 // code
 //