Extended examples/pybind11/use_dpctl_syclqueue/use_queue_device

Added a function that retrieves sub_group_sizes property of the device.

Author: oleksandr-pavlyk

examples/pybind11/use_dpctl_syclqueue/tests/test_queue_device.py

@@ -55,3 +55,11 @@ def test_offload_array_mod():
     Ynp = X % modulus_p
 
     assert np.array_equal(Y, Ynp)
+
+
+def test_get_sub_group_sizes():
+    d = dpctl.SyclDevice()
+    szs = uqd.get_sub_group_sizes(d)
+    assert type(szs) is list
+    assert all(type(el) is int for el in szs)
+    szs == d.sub_group_sizes

examples/pybind11/use_dpctl_syclqueue/use_queue_device/__init__.py

@@ -20,6 +20,7 @@
     get_device_global_mem_size,
     get_device_local_mem_size,
     get_max_compute_units,
+    get_sub_group_sizes,
     offloaded_array_mod,
 )
 
@@ -28,6 +29,7 @@
     "get_device_global_mem_size",
     "get_device_local_mem_size",
     "offloaded_array_mod",
+    "get_sub_group_sizes",
 ]
 
 __doc__ = """

examples/pybind11/use_dpctl_syclqueue/use_queue_device/_example.cpp

@@ -31,6 +31,7 @@
 #include <cstdint>
 #include <pybind11/numpy.h>
 #include <pybind11/pybind11.h>
+#include <pybind11/stl.h>
 
 namespace py = pybind11;
 
@@ -84,6 +85,11 @@ offloaded_array_mod(sycl::queue q,
     return res;
 }
 
+std::vector<std::size_t> get_sub_group_sizes(const sycl::device &d)
+{
+    return d.get_info<sycl::info::device::sub_group_sizes>();
+}
+
 PYBIND11_MODULE(_use_queue_device, m)
 {
     m.def(
@@ -100,4 +106,6 @@ PYBIND11_MODULE(_use_queue_device, m)
           "Computes amount of local memory of the given dpctl.SyclDevice");
     m.def("offloaded_array_mod", &offloaded_array_mod,
           "Compute offloaded modular reduction of integer-valued NumPy array");
+    m.def("get_sub_group_sizes", &get_sub_group_sizes,
+          "Gets info::device::sub_group_sizes property of given device");
 }