Skip to content

Commit d584ca6

Browse files
Add test_memset_fills_bytewise
1 parent 7240bad commit d584ca6

1 file changed

Lines changed: 16 additions & 0 deletions

File tree

dpctl/tests/test_sycl_queue_memset.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616

1717
"""Defines unit test cases for the SyclQueue.memset."""
1818

19+
import numpy as np
1920
import pytest
2021

2122
import dpctl
@@ -205,3 +206,18 @@ def test_memset_value_truncated_to_byte(val, expected):
205206
q.memset(mobj, val)
206207

207208
assert bytes(memoryview(mobj)) == bytes([expected]) * nbytes
209+
210+
211+
def test_memset_fills_bytewise():
212+
# memset fills byte-by-byte, so 0xAB reads back as 0xABABABAB per uint32
213+
try:
214+
q = dpctl.SyclQueue()
215+
except dpctl.SyclQueueCreationError:
216+
pytest.skip("Default constructor for SyclQueue failed")
217+
nelems = 16
218+
mobj = _create_memory(q, nelems * 4)
219+
220+
q.memset(mobj, 0xAB)
221+
222+
view = np.frombuffer(memoryview(mobj), dtype=np.uint32)
223+
assert np.all(view == 0xABABABAB)

0 commit comments

Comments
 (0)