Skip to content

Commit 5cdc287

Browse files
committed
Add a copy for strided arrays in dpnp.dot()
1 parent a7aa9af commit 5cdc287

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

dpnp/dpnp_iface_linearalgebra.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -106,9 +106,10 @@ def dot(x1, x2, out=None, **kwargs):
106106
# get USM type and queue to copy scalar from the host memory into a USM allocation
107107
usm_type, queue = get_usm_allocations([x1, x2]) if dpnp.isscalar(x1) or dpnp.isscalar(x2) else (None, None)
108108

109-
x1_desc = dpnp.get_dpnp_descriptor(x1, copy_when_strides=False, copy_when_nondefault_queue=False,
109+
# TODO: copy_when_strides=False (now it's done for faster implementation with transpose arrays)
110+
x1_desc = dpnp.get_dpnp_descriptor(x1, copy_when_strides=True, copy_when_nondefault_queue=False,
110111
alloc_usm_type=usm_type, alloc_queue=queue)
111-
x2_desc = dpnp.get_dpnp_descriptor(x2, copy_when_strides=False, copy_when_nondefault_queue=False,
112+
x2_desc = dpnp.get_dpnp_descriptor(x2, copy_when_strides=True, copy_when_nondefault_queue=False,
112113
alloc_usm_type=usm_type, alloc_queue=queue)
113114
if x1_desc and x2_desc:
114115
if out is not None:

0 commit comments

Comments
 (0)