File tree Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Original file line number Diff line number Diff line change 24
24
import dpctl .memory
25
25
26
26
27
- def _create_memory ():
27
+ def _create_memory (q ):
28
28
nbytes = 1024
29
- mobj = dpctl .memory .MemoryUSMShared (nbytes )
29
+ mobj = dpctl .memory .MemoryUSMShared (nbytes , queue = q )
30
30
return mobj
31
31
32
32
@@ -35,9 +35,9 @@ def _create_memory():
35
35
reason = "No SYCL devices except the default host device." ,
36
36
)
37
37
def test_memcpy_copy_usm_to_usm ():
38
- mobj1 = _create_memory ()
39
- mobj2 = _create_memory ()
40
38
q = dpctl .SyclQueue ()
39
+ mobj1 = _create_memory (q )
40
+ mobj2 = _create_memory (q )
41
41
42
42
mv1 = memoryview (mobj1 )
43
43
mv2 = memoryview (mobj2 )
@@ -54,8 +54,8 @@ def test_memcpy_copy_usm_to_usm():
54
54
# reason="No SYCL devices except the default host device."
55
55
# )
56
56
def test_memcpy_type_error ():
57
- mobj = _create_memory ()
58
- q = mobj . _queue
57
+ q = dpctl . SyclQueue ()
58
+ mobj = _create_memory ( q )
59
59
60
60
with pytest .raises (TypeError ) as cm :
61
61
q .memcpy (None , mobj , 3 )
You can’t perform that action at this time.
0 commit comments