@@ -15,7 +15,7 @@ def dprint(*args):
1515 sys .stdout .flush ()
1616
1717import dpctl
18- from dpctl ._memory import MemoryUSMShared
18+ from dpctl .memory import MemoryUSMShared
1919
2020functions_list = [o [0 ] for o in getmembers (np ) if isfunction (o [1 ]) or isbuiltin (o [1 ])]
2121class_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