Skip to content

Commit b5b0be0

Browse files
Use get_usm_type, get_usm_type_enum with other syclobj values
1 parent 4b472a6 commit b5b0be0

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

dpctl/tests/test_sycl_usm.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -271,8 +271,15 @@ def test_create_with_size_and_alignment(memory_ctor):
271271
reason="No SYCL devices except the default host device.",
272272
)
273273
def test_usm_type_execeptions():
274-
m = MemoryUSMDevice(1024)
274+
ctor = MemoryUSMDevice
275+
m = ctor(1024)
275276
assert m.nbytes == 1024
277+
q = m.sycl_queue
278+
assert m.get_usm_type(syclobj=q) == expected_usm_type_str(ctor)
279+
assert m.get_usm_type_enum(syclobj=q) == expected_usm_type_enum(ctor)
280+
ctx = q.sycl_context
281+
assert m.get_usm_type(syclobj=ctx) == expected_usm_type_str(ctor)
282+
assert m.get_usm_type_enum(syclobj=ctx) == expected_usm_type_enum(ctor)
276283
with pytest.raises(TypeError):
277284
m.get_usm_type(syclobj=Ellipsis)
278285
with pytest.raises(TypeError):

0 commit comments

Comments
 (0)