Skip to content

Commit f6d77fe

Browse files
Add square-matrix assertion in lu_solve
1 parent b68d80b commit f6d77fe

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

dpnp/linalg/dpnp_iface_linalg.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1037,6 +1037,7 @@ def lu_solve(lu_and_piv, b, trans=0, overwrite_b=False, check_finite=True):
10371037
(lu, piv) = lu_and_piv
10381038
dpnp.check_supported_arrays_type(lu, piv, b)
10391039
assert_stacked_2d(lu)
1040+
assert_stacked_square(lu)
10401041

10411042
return dpnp_lu_solve(
10421043
lu,

0 commit comments

Comments
 (0)