Skip to content

Commit 8a53212

Browse files
Adding test for Memory_GetOpaquePointer CAPI function
1 parent dc77858 commit 8a53212

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

dpctl/tests/test_sycl_usm.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -545,6 +545,7 @@ def test_cpython_api(memory_ctor):
545545
mod = sys.modules[mobj.__class__.__module__]
546546
# get capsules storing function pointers
547547
mem_ptr_fn_cap = mod.__pyx_capi__["Memory_GetUsmPointer"]
548+
mem_opaque_ptr_fn_cap = mod.__pyx_capi__["Memory_GetOpaquePointer"]
548549
mem_q_ref_fn_cap = mod.__pyx_capi__["Memory_GetQueueRef"]
549550
mem_ctx_ref_fn_cap = mod.__pyx_capi__["Memory_GetContextRef"]
550551
mem_nby_fn_cap = mod.__pyx_capi__["Memory_GetNumBytes"]
@@ -556,6 +557,9 @@ def test_cpython_api(memory_ctor):
556557
mem_ptr_fn_ptr = cap_ptr_fn(
557558
mem_ptr_fn_cap, b"DPCTLSyclUSMRef (struct Py_MemoryObject *)"
558559
)
560+
mem_opaque_ptr_fn_ptr = cap_ptr_fn(
561+
mem_opaque_ptr_fn_cap, b"void *(struct Py_MemoryObject *)"
562+
)
559563
mem_ctx_ref_fn_ptr = cap_ptr_fn(
560564
mem_ctx_ref_fn_cap, b"DPCTLSyclContextRef (struct Py_MemoryObject *)"
561565
)
@@ -571,6 +575,7 @@ def test_cpython_api(memory_ctor):
571575
)
572576
callable_maker = ctypes.PYFUNCTYPE(ctypes.c_void_p, ctypes.py_object)
573577
get_ptr_fn = callable_maker(mem_ptr_fn_ptr)
578+
get_opaque_ptr_fn = callable_maker(mem_opaque_ptr_fn_ptr)
574579
get_ctx_ref_fn = callable_maker(mem_ctx_ref_fn_ptr)
575580
get_q_ref_fn = callable_maker(mem_q_ref_fn_ptr)
576581
get_nby_fn = callable_maker(mem_nby_fn_ptr)
@@ -586,6 +591,8 @@ def test_cpython_api(memory_ctor):
586591
capi_ptr = get_ptr_fn(mobj)
587592
direct_ptr = mobj._pointer
588593
assert capi_ptr == direct_ptr
594+
capi_opaque_ptr = get_opaque_ptr_fn(mobj)
595+
assert capi_opaque_ptr != 0
589596
capi_ctx_ref = get_ctx_ref_fn(mobj)
590597
direct_ctx_ref = mobj._context.addressof_ref()
591598
assert capi_ctx_ref == direct_ctx_ref

0 commit comments

Comments
 (0)