Skip to content

Commit 7311b94

Browse files
fix comments
1 parent 6b892ea commit 7311b94

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

dpctl/dparray.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ def dprint(*args):
1515
sys.stdout.flush()
1616

1717
import dpctl
18-
from dpctl._memory import MemoryUSMShared
18+
from dpctl.memory import MemoryUSMShared
1919

2020
functions_list = [o[0] for o in getmembers(np) if isfunction(o[1]) or isbuiltin(o[1])]
2121
class_list = [o for o in getmembers(np) if isclass(o[1])]
@@ -38,7 +38,8 @@ def __new__(subtype, shape,
3838
nelems = np.prod(shape)
3939
dt = np.dtype(dtype)
4040
isz = dt.itemsize
41-
buf = MemoryUSMShared(nbytes=isz*max(1,nelems))
41+
nbytes = int(isz*max(1, nelems))
42+
buf = MemoryUSMShared(nbytes)
4243
new_obj = np.ndarray.__new__(
4344
subtype, shape, dtype=dt,
4445
buffer=buf, offset=0,
@@ -71,7 +72,8 @@ def __new__(subtype, shape,
7172
dtype=dtype, buffer=buffer,
7273
offset=offset, strides=strides,
7374
order=order)
74-
buf = MemoryUSMShared(nbytes=ar.nbytes)
75+
nbytes = int(ar.nbytes)
76+
buf = MemoryUSMShared(nbytes)
7577
new_obj = np.ndarray.__new__(
7678
subtype, shape, dtype=dtype,
7779
buffer=buf, offset=0,

0 commit comments

Comments
 (0)