Skip to content

Commit d267349

Browse files
Fix issue where dot works with incorrect shapes (#1111)
* Fix issue where dot works with incorrect shapes * Ceparate if condition and add x1_desc.ndim check Co-authored-by: Alexander-Makaryev <alexander.makaryev@gmail.com>
1 parent 67e933d commit d267349

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

dpnp/dpnp_iface_linearalgebra.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -98,6 +98,10 @@ def dot(x1, x2, **kwargs):
9898
if(x1_desc.ndim == 0 and x2_desc.strides is not None
9999
or x2_desc.ndim == 0 and x1_desc.strides is not None):
100100
pass
101+
elif (x1_desc.ndim >= 1 and x2_desc.ndim > 1 and x1_desc.shape[-1] != x2_desc.shape[-2]):
102+
pass
103+
elif (x1_desc.ndim > 0 and x2_desc.ndim == 1 and x1_desc.shape[-1] != x2_desc.shape[0]):
104+
pass
101105
else:
102106
return dpnp_dot(x1_desc, x2_desc).get_pyobj()
103107

0 commit comments

Comments
 (0)