Skip to content

Commit 0ed6e67

Browse files
Added dlpack exporter test for empty arrays.
1 parent 3cd9caf commit 0ed6e67

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed

dpctl/tests/test_usm_ndarray_dlpack.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,28 @@ def test_dlpack_exporter(typestr, usm_type):
8181
assert caps_fn(caps2, b"dltensor")
8282

8383

84+
def test_dlpack_exporter_empty(typestr, usm_type):
85+
caps_fn = ctypes.pythonapi.PyCapsule_IsValid
86+
caps_fn.restype = bool
87+
caps_fn.argtypes = [ctypes.py_object, ctypes.c_char_p]
88+
sycl_dev = dpctl.select_default_device()
89+
skip_if_dtype_not_supported(typestr, sycl_dev)
90+
X = dpt.empty((0,), dtype=typestr, usm_type=usm_type, device=sycl_dev)
91+
caps = X.__dlpack__()
92+
assert caps_fn(caps, b"dltensor")
93+
Y = dpt.empty(
94+
(
95+
1,
96+
0,
97+
),
98+
dtype=typestr,
99+
usm_type=usm_type,
100+
device=sycl_dev,
101+
)
102+
caps = Y.__dlpack__()
103+
assert caps_fn(caps, b"dltensor")
104+
105+
84106
def test_dlpack_exporter_stream():
85107
try:
86108
q1 = dpctl.SyclQueue()

0 commit comments

Comments
 (0)