5050import dpnp
5151import dpnp .backend .extensions .lapack ._lapack_impl as li
5252from dpnp .dpnp_utils import get_usm_allocations
53- from dpnp .linalg import LinAlgError as LinAlgError
5453
5554__all__ = [
5655 "assert_2d" ,
@@ -818,7 +817,7 @@ def _check_lapack_dev_info(dev_info, error_msg=None):
818817 if any (dev_info ):
819818 error_msg = error_msg or "Singular matrix"
820819
821- raise LinAlgError (error_msg )
820+ raise li . LinAlgError (error_msg )
822821
823822
824823def _common_type (* arrays ):
@@ -1737,7 +1736,7 @@ def assert_2d(*arrays):
17371736
17381737 for a in arrays :
17391738 if a .ndim != 2 :
1740- raise LinAlgError (
1739+ raise li . LinAlgError (
17411740 f"{ a .ndim } -dimensional array given. The input "
17421741 "array must be exactly two-dimensional"
17431742 )
@@ -1764,7 +1763,7 @@ def assert_stacked_2d(*arrays):
17641763
17651764 for a in arrays :
17661765 if a .ndim < 2 :
1767- raise LinAlgError (
1766+ raise li . LinAlgError (
17681767 f"{ a .ndim } -dimensional array given. The input "
17691768 "array must be at least two-dimensional"
17701769 )
@@ -1800,7 +1799,7 @@ def assert_stacked_square(*arrays):
18001799 for a in arrays :
18011800 m , n = a .shape [- 2 :]
18021801 if m != n :
1803- raise LinAlgError (
1802+ raise li . LinAlgError (
18041803 "Last 2 dimensions of the input array must be square"
18051804 )
18061805
@@ -1944,7 +1943,7 @@ def dpnp_cond(x, p=None):
19441943 """Compute the condition number of a matrix."""
19451944
19461945 if _is_empty_2d (x ):
1947- raise LinAlgError ("cond is not defined on empty arrays" )
1946+ raise li . LinAlgError ("cond is not defined on empty arrays" )
19481947 if p is None or p == 2 or p == - 2 :
19491948 s = dpnp .linalg .svd (x , compute_uv = False )
19501949 if p == - 2 :
@@ -2198,15 +2197,15 @@ def dpnp_lstsq(a, b, rcond=None):
21982197 """
21992198
22002199 if b .ndim > 2 :
2201- raise LinAlgError (
2200+ raise li . LinAlgError (
22022201 f"{ b .ndim } -dimensional array given. The input "
22032202 "array must be exactly two-dimensional"
22042203 )
22052204
22062205 m , n = a .shape [- 2 :]
22072206 m2 = b .shape [0 ]
22082207 if m != m2 :
2209- raise LinAlgError ("Incompatible dimensions" )
2208+ raise li . LinAlgError ("Incompatible dimensions" )
22102209
22112210 u , s , vh = dpnp_svd (a , full_matrices = False , related_arrays = [b ])
22122211
@@ -2318,20 +2317,20 @@ def dpnp_multi_dot(n, arrays, out=None):
23182317 """Compute dot product of two or more arrays in a single function call."""
23192318
23202319 if not arrays [0 ].ndim in [1 , 2 ]:
2321- raise LinAlgError (
2320+ raise li . LinAlgError (
23222321 f"{ arrays [0 ].ndim } -dimensional array given. "
23232322 "First array must be 1-D or 2-D."
23242323 )
23252324
23262325 if not arrays [- 1 ].ndim in [1 , 2 ]:
2327- raise LinAlgError (
2326+ raise li . LinAlgError (
23282327 f"{ arrays [- 1 ].ndim } -dimensional array given. "
23292328 "Last array must be 1-D or 2-D."
23302329 )
23312330
23322331 for arr in arrays [1 :- 1 ]:
23332332 if arr .ndim != 2 :
2334- raise LinAlgError (
2333+ raise li . LinAlgError (
23352334 f"{ arr .ndim } -dimensional array given. Inner arrays must be 2-D."
23362335 )
23372336
0 commit comments