fix tolerance
@@ -511,9 +511,6 @@ class TestPolyArithmeticDiffTypes(unittest.TestCase):
'rcond': [None, 0.5, 1e-15],
'weighted': [True, False]
}))
-@pytest.mark.skipif(
- cupy.cuda.runtime.runtimeGetVersion() < 10000,
- reason='polyfit has issues with CUDA 9.x')
class TestPolyfitParametersCombinations(unittest.TestCase):
def _full_fit(self, xp, dtype):
@@ -538,7 +535,7 @@ class TestPolyfitParametersCombinations(unittest.TestCase):
testing.assert_allclose(cp_c, np_c, atol=1e-9)
testing.assert_allclose(cp_resids, np_resids, atol=1e-9)
- testing.assert_allclose(cp_s, np_s)
+ testing.assert_allclose(cp_s, np_s, atol=1e-9)
assert cp_rank == np_rank
if self.rcond is not None:
assert cp_rcond == np_rcond
