File tree Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Original file line number Diff line number Diff line change @@ -271,8 +271,15 @@ def test_create_with_size_and_alignment(memory_ctor):
271
271
reason = "No SYCL devices except the default host device." ,
272
272
)
273
273
def test_usm_type_execeptions ():
274
- m = MemoryUSMDevice (1024 )
274
+ ctor = MemoryUSMDevice
275
+ m = ctor (1024 )
275
276
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 )
276
283
with pytest .raises (TypeError ):
277
284
m .get_usm_type (syclobj = Ellipsis )
278
285
with pytest .raises (TypeError ):
You can’t perform that action at this time.
0 commit comments