Skip to content

Commit e2185e9

Browse files
Fixes #649
``` In [1]: import dpctl.tensor as dpt, itertools In [2]: a = dpt.usm_ndarray((2,3)) In [3]: for i,j in itertools.product(range(2), range(3)): a[i, j] = i*3 + j In [4]: dpt.to_numpy(a)[1:].T Out[4]: array([[3.], [4.], [5.]]) In [5]: dpt.to_numpy(a[1:].T) Out[5]: array([[3.], [4.], [5.]]) ```
1 parent 3d5d886 commit e2185e9

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

dpctl/tensor/_usmarray.pyx

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1054,7 +1054,8 @@ cdef usm_ndarray _transpose(usm_ndarray ary):
10541054
_make_reversed_int_tuple(ary.nd_, ary.strides_)
10551055
if (ary.strides_) else None),
10561056
buffer=ary.base_,
1057-
order=('F' if (ary.flags_ & USM_ARRAY_C_CONTIGUOUS) else 'C')
1057+
order=('F' if (ary.flags_ & USM_ARRAY_C_CONTIGUOUS) else 'C'),
1058+
offset=ary.get_offset()
10581059
)
10591060
r.flags_ |= (ary.flags_ & USM_ARRAY_WRITEABLE)
10601061
return r

0 commit comments

Comments
 (0)