Added example of getting partition_max_sub_devices using pybind11

Author: oleksandr-pavlyk

examples/pybind11/use_dpctl_syclqueue/tests/test_queue_device.py

@@ -63,3 +63,11 @@ def test_get_sub_group_sizes():
     assert type(szs) is list
     assert all(type(el) is int for el in szs)
     szs == d.sub_group_sizes
+
+
+def test_get_partition_max_sub_devices():
+    d = dpctl.SyclDevice()
+    mt = uqd.get_partition_max_sub_devices(d)
+    assert type(mt) is int
+    assert mt >= 0
+    assert mt <= d.max_compute_units

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_partition_max_sub_devices,
     get_sub_group_sizes,
     offloaded_array_mod,
 )
@@ -30,6 +31,7 @@
     "get_device_local_mem_size",
     "offloaded_array_mod",
     "get_sub_group_sizes",
+    "get_partition_max_sub_devices",
 ]
 
 __doc__ = """

examples/pybind11/use_dpctl_syclqueue/use_queue_device/_example.cpp

@@ -90,6 +90,11 @@ std::vector<std::size_t> get_sub_group_sizes(const sycl::device &d)
     return d.get_info<sycl::info::device::sub_group_sizes>();
 }
 
+std::uint32_t get_partition_max_sub_devices(const sycl::device &d)
+{
+    return d.get_info<sycl::info::device::partition_max_sub_devices>();
+}
+
 PYBIND11_MODULE(_use_queue_device, m)
 {
     m.def(
@@ -108,4 +113,7 @@ PYBIND11_MODULE(_use_queue_device, m)
           "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");
+    m.def("get_partition_max_sub_devices", &get_partition_max_sub_devices,
+          "Gets info::device::partition_max_sub_devices for given "
+          "dpctl.SyclDevice");
 }