Skip to content

Commit 5165cc0

Browse files
FIX: additional linreg GPU threshold upgrade (#1501) (#1516)
(cherry picked from commit a43c257) Co-authored-by: ethanglaser <42726565+ethanglaser@users.noreply.github.com>
1 parent 23c6be1 commit 5165cc0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

onedal/linear_model/tests/test_linear_regression.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ def test_full_results(queue, dtype):
7777
model.fit(X, y, queue=queue)
7878

7979
if queue.sycl_device.is_gpu:
80-
tol = 2.5e-3 if model.coef_.dtype == np.float32 else 1e-5
80+
tol = 5e-3 if model.coef_.dtype == np.float32 else 1e-5
8181
else:
8282
tol = 2e-3 if model.coef_.dtype == np.float32 else 1e-5
8383
assert_allclose(coef, model.coef_.T, rtol=tol)

0 commit comments

Comments
 (0)