FIX: additional linreg GPU threshold upgrade (#1501) (#1516)

(cherry picked from commit a43c257) Co-authored-by: ethanglaser <[email protected]>
uxlfoundation · Sep 28, 2023 · 5165cc0 · 5165cc0
1 parent 23c6be1
commit 5165cc0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onedal/linear_model/tests/test_linear_regression.py b/onedal/linear_model/tests/test_linear_regression.py
@@ -77,7 +77,7 @@ def test_full_results(queue, dtype):
         model.fit(X, y, queue=queue)
 
         if queue.sycl_device.is_gpu:
-            tol = 2.5e-3 if model.coef_.dtype == np.float32 else 1e-5
+            tol = 5e-3 if model.coef_.dtype == np.float32 else 1e-5
         else:
             tol = 2e-3 if model.coef_.dtype == np.float32 else 1e-5
         assert_allclose(coef, model.coef_.T, rtol=tol)