Added size_t *DPCTLDevice_GetSubGroupSizes(DRef, size_t res_len)

The function exposes `device::get_info<info::device::sub_group_sizes>()`
which returns `std::vector<size_t>`.

DPCTLDevice_GetSubGroupSizes returns pointer to allocated array, populated
with the content of the result std::vector. res_len is set with the size
of the result std::vector.

Author: oleksandr-pavlyk

libsyclinterface/include/dpctl_sycl_device_interface.h

@@ -651,4 +651,17 @@ DPCTL_API
 DPCTLGlobalMemCacheType
 DPCTLDevice_GetGlobalMemCacheType(__dpctl_keep const DPCTLSyclDeviceRef DRef);
 
+/*!
+ * @brief Wrapper for get_info<info::device::sub_group_sizes>().
+ *
+ * @param    DRef           Opaque pointer to a ``sycl::device``
+ * @param    res_len        Populated with size of the returned array
+ * @return   Returns the valid result if device exists else returns NULL.
+ * @ingroup DeviceInterface
+ */
+DPCTL_API
+__dpctl_keep size_t *
+DPCTLDevice_GetSubGroupSizes(__dpctl_keep const DPCTLSyclDeviceRef DRef,
+                             size_t *res_len);
+
 DPCTL_C_EXTERN_C_END

libsyclinterface/source/dpctl_sycl_device_interface.cpp

@@ -743,3 +743,30 @@ DPCTLDevice_GetGlobalMemCacheType(__dpctl_keep const DPCTLSyclDeviceRef DRef)
         return DPCTL_MEM_CACHE_TYPE_INDETERMINATE;
     }
 }
+
+__dpctl_keep size_t *
+DPCTLDevice_GetSubGroupSizes(__dpctl_keep const DPCTLSyclDeviceRef DRef,
+                             size_t *res_len)
+{
+    size_t *sizes = nullptr;
+    std::vector<size_t> sg_sizes;
+    *res_len = 0;
+    auto D = unwrap<device>(DRef);
+    if (D) {
+        try {
+            sg_sizes = D->get_info<info::device::sub_group_sizes>();
+            *res_len = sg_sizes.size();
+        } catch (std::exception const &e) {
+            error_handler(e, __FILE__, __func__, __LINE__);
+        }
+        try {
+            sizes = new size_t[sg_sizes.size()];
+        } catch (std::exception const &e) {
+            error_handler(e, __FILE__, __func__, __LINE__);
+        }
+        for (auto i = 0ul; (sizes != nullptr) && i < sg_sizes.size(); ++i) {
+            sizes[i] = sg_sizes[i];
+        }
+    }
+    return sizes;
+}

libsyclinterface/tests/test_sycl_device_interface.cpp

@@ -205,6 +205,22 @@ TEST_P(TestDPCTLSyclDeviceInterface, ChkGetMaxNumSubGroups)
         EXPECT_TRUE(n > 0);
 }
 
+TEST_P(TestDPCTLSyclDeviceInterface, ChkGetSubGroupSizes)
+{
+    size_t sg_sizes_len = 0;
+    size_t *sg_sizes = nullptr;
+    EXPECT_NO_FATAL_FAILURE(
+        sg_sizes = DPCTLDevice_GetSubGroupSizes(DRef, &sg_sizes_len));
+    if (DPCTLDevice_IsAccelerator(DRef))
+        EXPECT_TRUE(sg_sizes_len >= 0);
+    else
+        EXPECT_TRUE(sg_sizes_len > 0);
+    for (size_t i = 0; i < sg_sizes_len; ++i) {
+        EXPECT_TRUE(sg_sizes > 0);
+    }
+    EXPECT_NO_FATAL_FAILURE(DPCTLSize_t_Array_Delete(sg_sizes));
+}
+
 TEST_P(TestDPCTLSyclDeviceInterface, ChkGetPlatform)
 {
     DPCTLSyclPlatformRef PRef = nullptr;
@@ -751,3 +767,13 @@ TEST_F(TestDPCTLSyclDeviceNullArgs, ChkGetGlobalMemCacheType)
     EXPECT_NO_FATAL_FAILURE(res = DPCTLDevice_GetGlobalMemCacheType(Null_DRef));
     ASSERT_TRUE(res == DPCTL_MEM_CACHE_TYPE_INDETERMINATE);
 }
+
+TEST_F(TestDPCTLSyclDeviceNullArgs, ChkGetSubGroupSizes)
+{
+    size_t *sg_sizes = nullptr;
+    size_t sg_sizes_len = 0;
+    EXPECT_NO_FATAL_FAILURE(
+        sg_sizes = DPCTLDevice_GetSubGroupSizes(Null_DRef, &sg_sizes_len));
+    ASSERT_TRUE(sg_sizes == nullptr);
+    ASSERT_TRUE(sg_sizes_len == 0);
+}