@@ -1935,7 +1935,7 @@ def test_matrix_rank(self, data, dtype):
1935
1935
1936
1936
np_rank = numpy .linalg .matrix_rank (a )
1937
1937
dp_rank = dpnp .linalg .matrix_rank (a_dp )
1938
- assert np_rank == dp_rank
1938
+ assert dp_rank . asnumpy () == np_rank
1939
1939
1940
1940
@pytest .mark .parametrize ("dtype" , get_all_dtypes ())
1941
1941
@pytest .mark .parametrize (
@@ -1953,7 +1953,7 @@ def test_matrix_rank_hermitian(self, data, dtype):
1953
1953
1954
1954
np_rank = numpy .linalg .matrix_rank (a , hermitian = True )
1955
1955
dp_rank = dpnp .linalg .matrix_rank (a_dp , hermitian = True )
1956
- assert np_rank == dp_rank
1956
+ assert dp_rank . asnumpy () == np_rank
1957
1957
1958
1958
@pytest .mark .parametrize (
1959
1959
"high_tol, low_tol" ,
@@ -1986,15 +1986,15 @@ def test_matrix_rank_tolerance(self, high_tol, low_tol):
1986
1986
dp_rank_high_tol = dpnp .linalg .matrix_rank (
1987
1987
a_dp , hermitian = True , tol = dp_high_tol
1988
1988
)
1989
- assert np_rank_high_tol == dp_rank_high_tol
1989
+ assert dp_rank_high_tol . asnumpy () == np_rank_high_tol
1990
1990
1991
1991
np_rank_low_tol = numpy .linalg .matrix_rank (
1992
1992
a , hermitian = True , tol = low_tol
1993
1993
)
1994
1994
dp_rank_low_tol = dpnp .linalg .matrix_rank (
1995
1995
a_dp , hermitian = True , tol = dp_low_tol
1996
1996
)
1997
- assert np_rank_low_tol == dp_rank_low_tol
1997
+ assert dp_rank_low_tol . asnumpy () == np_rank_low_tol
1998
1998
1999
1999
# rtol kwarg was added in numpy 2.0
2000
2000
@testing .with_requires ("numpy>=2.0" )
@@ -2789,15 +2789,14 @@ def check_decomposition(
2789
2789
for i in range (min (dp_a .shape [- 2 ], dp_a .shape [- 1 ])):
2790
2790
dpnp_diag_s [..., i , i ] = dp_s [..., i ]
2791
2791
reconstructed = dpnp .dot (dp_u , dpnp .dot (dpnp_diag_s , dp_vt ))
2792
- # TODO: use assert dpnp.allclose() inside check_decomposition()
2793
- # when it will support complex dtypes
2794
- assert_allclose (dp_a , reconstructed , rtol = tol , atol = 1e-4 )
2792
+
2793
+ assert dpnp .allclose (dp_a , reconstructed , rtol = tol , atol = 1e-4 )
2795
2794
2796
2795
assert_allclose (dp_s , np_s , rtol = tol , atol = 1e-03 )
2797
2796
2798
2797
if compute_vt :
2799
2798
for i in range (min (dp_a .shape [- 2 ], dp_a .shape [- 1 ])):
2800
- if np_u [..., 0 , i ] * dp_u [..., 0 , i ] < 0 :
2799
+ if np_u [..., 0 , i ] * dpnp . asnumpy ( dp_u [..., 0 , i ]) < 0 :
2801
2800
np_u [..., :, i ] = - np_u [..., :, i ]
2802
2801
np_vt [..., i , :] = - np_vt [..., i , :]
2803
2802
for i in range (numpy .count_nonzero (np_s > tol )):
0 commit comments