We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c9dcbf4 commit da4ce75Copy full SHA for da4ce75
1 file changed
examples/pybind11/external_usm_allocation/tests/test_direct.py
@@ -17,10 +17,10 @@
17
# coding: utf-8
18
19
import external_usm_allocation as eua
20
+import numpy as np
21
22
import dpctl
23
import dpctl.memory as dpm
-import dpctl.tensor as dpt
24
25
26
def test_direct():
@@ -30,8 +30,11 @@ def test_direct():
30
mbuf = eua.make_zeroed_device_memory(nb, q)
31
32
assert isinstance(mbuf, dpm.MemoryUSMDevice)
33
- assert mbuf.nbytes == 2 * 30
+ assert mbuf.nbytes == nb
34
assert mbuf.sycl_queue == q
35
36
- x = dpt.usm_ndarray(30, dtype="i2", buffer=mbuf)
37
- assert dpt.all(x == dpt.zeros(30, dtype="i2", sycl_queue=q))
+ x = np.empty(30, dtype="i2")
+ assert x.nbytes == nb
38
+
39
+ q.memcpy(dest=x, src=mbuf, count=nb)
40
+ assert np.all(x == 0)
0 commit comments