Skip to content

Commit 00bd38a

Browse files
Be explicit about allocation queue in tests
1 parent 4f19e68 commit 00bd38a

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

dpctl/tests/test_sycl_queue_memcpy.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -24,9 +24,9 @@
2424
import dpctl.memory
2525

2626

27-
def _create_memory():
27+
def _create_memory(q):
2828
nbytes = 1024
29-
mobj = dpctl.memory.MemoryUSMShared(nbytes)
29+
mobj = dpctl.memory.MemoryUSMShared(nbytes, queue=q)
3030
return mobj
3131

3232

@@ -35,9 +35,9 @@ def _create_memory():
3535
reason="No SYCL devices except the default host device.",
3636
)
3737
def test_memcpy_copy_usm_to_usm():
38-
mobj1 = _create_memory()
39-
mobj2 = _create_memory()
4038
q = dpctl.SyclQueue()
39+
mobj1 = _create_memory(q)
40+
mobj2 = _create_memory(q)
4141

4242
mv1 = memoryview(mobj1)
4343
mv2 = memoryview(mobj2)
@@ -54,8 +54,8 @@ def test_memcpy_copy_usm_to_usm():
5454
# reason="No SYCL devices except the default host device."
5555
# )
5656
def test_memcpy_type_error():
57-
mobj = _create_memory()
58-
q = mobj._queue
57+
q = dpctl.SyclQueue()
58+
mobj = _create_memory(q)
5959

6060
with pytest.raises(TypeError) as cm:
6161
q.memcpy(None, mobj, 3)

0 commit comments

Comments
 (0)