@@ -32,14 +32,14 @@ def get_spirv_abspath(fn):
3232 return spirv_file
3333
3434
35- def _check_cpython_api_SyclKernelBundle_GetKernelBundleRef (sycl_prog ):
35+ def _check_cpython_api_SyclKernelBundle_GetKernelBundleRef (kb ):
3636 """Checks Cython-generated C-API function
37- `SyclKernelBundle_GetKernelBundleRef` defined in _program .pyx"""
37+ `SyclKernelBundle_GetKernelBundleRef` defined in _compiler .pyx"""
3838 import ctypes
3939 import sys
4040
41- assert type (sycl_prog ) is dpc .SyclKernelBundle
42- mod = sys .modules [sycl_prog .__class__ .__module__ ]
41+ assert type (kb ) is dpc .SyclKernelBundle
42+ mod = sys .modules [kb .__class__ .__module__ ]
4343 # get capsule storing SyclKernelBundle_GetKernelBundleRef function ptr
4444 kb_ref_fn_cap = mod .__pyx_capi__ ["SyclKernelBundle_GetKernelBundleRef" ]
4545 # construct Python callable to invoke "SyclKernelBundle_GetKernelBundleRef"
@@ -54,41 +54,41 @@ def _check_cpython_api_SyclKernelBundle_GetKernelBundleRef(sycl_prog):
5454 callable_maker = ctypes .PYFUNCTYPE (ctypes .c_void_p , ctypes .py_object )
5555 get_kernel_bundle_ref_fn = callable_maker (kb_ref_fn_ptr )
5656
57- r2 = sycl_prog .addressof_ref ()
58- r1 = get_kernel_bundle_ref_fn (sycl_prog )
57+ r2 = kb .addressof_ref ()
58+ r1 = get_kernel_bundle_ref_fn (kb )
5959 assert r1 == r2
6060
6161
62- def _check_cpython_api_SyclKernelBundle_Make (sycl_prog ):
62+ def _check_cpython_api_SyclKernelBundle_Make (kb ):
6363 """Checks Cython-generated C-API function
64- `SyclKernelBundle_Make` defined in _program .pyx"""
64+ `SyclKernelBundle_Make` defined in _compiler .pyx"""
6565 import ctypes
6666 import sys
6767
68- assert type (sycl_prog ) is dpc .SyclKernelBundle
69- mod = sys .modules [sycl_prog .__class__ .__module__ ]
68+ assert type (kb ) is dpc .SyclKernelBundle
69+ mod = sys .modules [kb .__class__ .__module__ ]
7070 # get capsule storing SyclKernelBundle_Make function ptr
71- make_prog_fn_cap = mod .__pyx_capi__ ["SyclKernelBundle_Make" ]
71+ make_kb_fn_cap = mod .__pyx_capi__ ["SyclKernelBundle_Make" ]
7272 # construct Python callable to invoke "SyclKernelBundle_Make"
7373 cap_ptr_fn = ctypes .pythonapi .PyCapsule_GetPointer
7474 cap_ptr_fn .restype = ctypes .c_void_p
7575 cap_ptr_fn .argtypes = [ctypes .py_object , ctypes .c_char_p ]
76- make_prog_fn_ptr = cap_ptr_fn (
77- make_prog_fn_cap ,
76+ make_kb_fn_ptr = cap_ptr_fn (
77+ make_kb_fn_cap ,
7878 b"struct PySyclKernelBundleObject *(DPCTLSyclKernelBundleRef)" ,
7979 )
8080 # PYFUNCTYPE(result_type, *arg_types)
8181 callable_maker = ctypes .PYFUNCTYPE (ctypes .py_object , ctypes .c_void_p )
82- make_prog_fn = callable_maker (make_prog_fn_ptr )
82+ make_kb_fn = callable_maker (make_kb_fn_ptr )
8383
84- p2 = make_prog_fn ( sycl_prog .addressof_ref ())
85- assert p2 .has_sycl_kernel ("add" )
86- assert p2 .has_sycl_kernel ("axpy" )
84+ kb2 = make_kb_fn ( kb .addressof_ref ())
85+ assert kb2 .has_sycl_kernel ("add" )
86+ assert kb2 .has_sycl_kernel ("axpy" )
8787
8888
8989def _check_cpython_api_SyclKernel_GetKernelRef (krn ):
9090 """Checks Cython-generated C-API function
91- `SyclKernel_GetKernelRef` defined in _program .pyx"""
91+ `SyclKernel_GetKernelRef` defined in _compiler .pyx"""
9292 import ctypes
9393 import sys
9494
@@ -114,7 +114,7 @@ def _check_cpython_api_SyclKernel_GetKernelRef(krn):
114114
115115def _check_cpython_api_SyclKernel_Make (krn ):
116116 """Checks Cython-generated C-API function
117- `SyclKernel_Make` defined in _program .pyx"""
117+ `SyclKernel_Make` defined in _compiler .pyx"""
118118 import ctypes
119119 import sys
120120
@@ -149,7 +149,7 @@ def _check_cpython_api_SyclKernel_Make(krn):
149149 assert krn .work_group_size == k3 .work_group_size
150150
151151
152- def _check_multi_kernel_program (kb ):
152+ def _check_multi_kernel_bundle (kb ):
153153 assert type (kb ) is dpc .SyclKernelBundle
154154
155155 assert type (kb .addressof_ref ()) is int
@@ -206,7 +206,7 @@ def test_create_kernel_bundle_from_source_ocl():
206206 except dpctl .SyclQueueCreationError :
207207 pytest .skip ("No OpenCL queue is available" )
208208 kb = dpc .create_kernel_bundle_from_source (q , oclSrc )
209- _check_multi_kernel_program (kb )
209+ _check_multi_kernel_bundle (kb )
210210
211211
212212def test_create_kernel_bundle_from_spirv_ocl ():
@@ -218,7 +218,7 @@ def test_create_kernel_bundle_from_spirv_ocl():
218218 with open (spirv_file , "rb" ) as fin :
219219 spirv = fin .read ()
220220 kb = dpc .create_kernel_bundle_from_spirv (q , spirv )
221- _check_multi_kernel_program (kb )
221+ _check_multi_kernel_bundle (kb )
222222
223223
224224def test_create_kernel_bundle_from_spirv_l0 ():
@@ -230,7 +230,7 @@ def test_create_kernel_bundle_from_spirv_l0():
230230 with open (spirv_file , "rb" ) as fin :
231231 spirv = fin .read ()
232232 kb = dpc .create_kernel_bundle_from_spirv (q , spirv )
233- _check_multi_kernel_program (kb )
233+ _check_multi_kernel_bundle (kb )
234234
235235
236236@pytest .mark .xfail (
@@ -251,7 +251,7 @@ def test_create_kernel_bundle_from_source_l0():
251251 c[index] = a[index] + d*b[index]; \
252252 }"
253253 kb = dpc .create_kernel_bundle_from_source (q , oclSrc )
254- _check_multi_kernel_program (kb )
254+ _check_multi_kernel_bundle (kb )
255255
256256
257257def test_create_kernel_bundle_from_invalid_src_ocl ():
0 commit comments