Implemented SyclKernel.max_sub_group_size

Author: oleksandr-pavlyk

dpctl/_backend.pxd

@@ -271,8 +271,7 @@ cdef extern from "syclinterface/dpctl_sycl_kernel_interface.h":
     cdef size_t DPCTLKernel_GetPreferredWorkGroupSizeMultiple(const DPCTLSyclKernelRef KRef)
     cdef size_t DPCTLKernel_GetPrivateMemSize(const DPCTLSyclKernelRef KRef)
     cdef uint32_t DPCTLKernel_GetMaxNumSubGroups(const DPCTLSyclKernelRef KRef)
-##  Next line is commented out due to issue in DPC++ runtime
-#   cdef uint32_t DPCTLKernel_GetMaxSubGroupSize(const DPCTLSyclKernelRef KRef)
+    cdef uint32_t DPCTLKernel_GetMaxSubGroupSize(const DPCTLSyclKernelRef KRef)
     cdef uint32_t DPCTLKernel_GetCompileNumSubGroups(const DPCTLSyclKernelRef KRef)
     cdef uint32_t DPCTLKernel_GetCompileSubGroupSize(const DPCTLSyclKernelRef KRef)
 

dpctl/program/_program.pyx

@@ -34,6 +34,7 @@ from dpctl._backend cimport (  # noqa: E211, E402;
     DPCTLKernel_GetCompileNumSubGroups,
     DPCTLKernel_GetCompileSubGroupSize,
     DPCTLKernel_GetMaxNumSubGroups,
+    DPCTLKernel_GetMaxSubGroupSize,
     DPCTLKernel_GetNumArgs,
     DPCTLKernel_GetPreferredWorkGroupSizeMultiple,
     DPCTLKernel_GetPrivateMemSize,
@@ -146,8 +147,8 @@ cdef class SyclKernel:
     def max_sub_group_size(self):
         """ Returns the maximum sub-groups size for this kernel.
         """
-        cdef uint32_t sz = 0
-        return NotImplemented
+        cdef uint32_t sz = DPCTLKernel_GetMaxSubGroupSize(self._kernel_ref)
+        return sz
 
     @property
     def compile_num_sub_groups(self):

dpctl/tests/test_sycl_program.py

@@ -180,9 +180,7 @@ def _check_multi_kernel_program(prog):
         vmnsg = krn.max_num_sub_groups
         assert type(vmnsg) is int
         v = krn.max_sub_group_size
-        assert (
-            v == NotImplemented
-        ), "SyclKernel.max_sub_group_size acquired implementation, fix the test"
+        assert type(v) is int
         cmnsg = krn.compile_num_sub_groups
         assert type(cmnsg) is int
         cmsgsz = krn.compile_sub_group_size