Merge pull request #7966 from andfoy/fix_solve_lapack

Import cupyx.lapack inside cupy.linalg.solve

cupy/linalg/_solve.py

@@ -35,6 +35,7 @@ def solve(a, b):
 
     .. seealso:: :func:`numpy.linalg.solve`
     """
+    from cupyx import lapack
     from cupy.cublas import batched_gesv, get_batched_gesv_limit
 
     if a.ndim > 2 and a.shape[-1] <= get_batched_gesv_limit():
@@ -64,7 +65,7 @@ def solve(a, b):
         # prevent 'a' and 'b' to be overwritten
         a = a.astype(dtype, copy=True, order='F')
         b = b.astype(dtype, copy=True, order='F')
-        cupyx.lapack.gesv(a, b)
+        lapack.gesv(a, b)
         return b.astype(out_dtype, copy=False)
 
     # prevent 'a' to be overwritten
@@ -75,7 +76,7 @@ def solve(a, b):
         index = numpy.unravel_index(i, shape)
         # prevent 'b' to be overwritten
         bi = b[index].astype(dtype, copy=True, order='F')
-        cupyx.lapack.gesv(a[index], bi)
+        lapack.gesv(a[index], bi)
         x[index] = bi
     return x