@@ -2182,10 +2182,7 @@ def test_lu_solve(self, shape, rhs_cols, order, dtype):
2182
2182
2183
2183
# check A @ x = b
2184
2184
Ax = a_dp @ x
2185
- if dpnp .issubdtype (dtype , dpnp .integer ):
2186
- assert dpnp .allclose (Ax , b_dp , rtol = 1e-5 , atol = 1e-5 )
2187
- else :
2188
- assert dpnp .allclose (Ax , b_dp , rtol = 1e-6 , atol = 1e-6 )
2185
+ assert dpnp .allclose (Ax , b_dp , rtol = 1e-5 , atol = 1e-5 )
2189
2186
2190
2187
@pytest .mark .parametrize ("trans" , [0 , 1 , 2 ])
2191
2188
@pytest .mark .parametrize ("dtype" , get_float_complex_dtypes ())
@@ -2207,7 +2204,7 @@ def test_trans(self, trans, dtype):
2207
2204
else : # trans == 2
2208
2205
lhs = a_dp .conj ().T @ x
2209
2206
2210
- assert dpnp .allclose (lhs , b_dp , rtol = 1e-6 , atol = 1e-6 )
2207
+ assert dpnp .allclose (lhs , b_dp , rtol = 1e-5 , atol = 1e-5 )
2211
2208
2212
2209
@pytest .mark .parametrize ("dtype" , get_float_complex_dtypes ())
2213
2210
def test_overwrite_inplace (self , dtype ):
@@ -2272,7 +2269,7 @@ def test_diff_type(self, dtype_a, dtype_b):
2272
2269
lu , piv = dpnp .linalg .lu_factor (a_dp , check_finite = False )
2273
2270
x = dpnp .linalg .lu_solve ((lu , piv ), b_dp , check_finite = False )
2274
2271
assert dpnp .allclose (
2275
- a_dp @ x , b_dp .astype (x .dtype , copy = False ), rtol = 1e-6 , atol = 1e-6
2272
+ a_dp @ x , b_dp .astype (x .dtype , copy = False ), rtol = 1e-5 , atol = 1e-5
2276
2273
)
2277
2274
2278
2275
def test_strided_rhs (self ):
@@ -2295,7 +2292,7 @@ def test_strided_rhs(self):
2295
2292
(lu , piv ), b_dp , overwrite_b = False , check_finite = False
2296
2293
)
2297
2294
2298
- assert dpnp .allclose (a_dp @ x , b_dp , rtol = 1e-6 , atol = 1e-6 )
2295
+ assert dpnp .allclose (a_dp @ x , b_dp , rtol = 1e-5 , atol = 1e-5 )
2299
2296
2300
2297
@pytest .mark .parametrize (
2301
2298
"b_shape" ,
@@ -2324,7 +2321,7 @@ def test_broadcast_rhs(self, b_shape):
2324
2321
2325
2322
assert x .shape == b_dp .shape
2326
2323
2327
- assert dpnp .allclose (a_dp @ x , b_dp , rtol = 1e-6 , atol = 1e-6 )
2324
+ assert dpnp .allclose (a_dp @ x , b_dp , rtol = 1e-5 , atol = 1e-5 )
2328
2325
2329
2326
@pytest .mark .parametrize ("shape" , [(0 , 0 ), (0 , 5 ), (5 , 5 )])
2330
2327
@pytest .mark .parametrize ("rhs_cols" , [None , 0 , 3 ])
0 commit comments