File tree Expand file tree Collapse file tree 1 file changed +10
-2
lines changed Expand file tree Collapse file tree 1 file changed +10
-2
lines changed Original file line number Diff line number Diff line change @@ -300,14 +300,22 @@ def _copy_from_usm_ndarray_to_usm_ndarray(dst, src):
300
300
src .shape , src .strides , len (common_shape )
301
301
)
302
302
src_same_shape = dpt .usm_ndarray (
303
- common_shape , dtype = src .dtype , buffer = src , strides = new_src_strides
303
+ common_shape ,
304
+ dtype = src .dtype ,
305
+ buffer = src ,
306
+ strides = new_src_strides ,
307
+ offset = src ._element_offset ,
304
308
)
305
309
elif src .ndim == len (common_shape ):
306
310
new_src_strides = _broadcast_strides (
307
311
src .shape , src .strides , len (common_shape )
308
312
)
309
313
src_same_shape = dpt .usm_ndarray (
310
- common_shape , dtype = src .dtype , buffer = src , strides = new_src_strides
314
+ common_shape ,
315
+ dtype = src .dtype ,
316
+ buffer = src ,
317
+ strides = new_src_strides ,
318
+ offset = src ._element_offset ,
311
319
)
312
320
else :
313
321
# since broadcasting succeeded, src.ndim is greater because of
You can’t perform that action at this time.
0 commit comments