Skip to content

Commit 9726e1b

Browse files
gneculajax authors
authored andcommitted
Adjust tolerance for backward compatibility test for Qr on TPU.
Fixes flaky test. PiperOrigin-RevId: 575752219
1 parent c569f8e commit 9726e1b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

jax/experimental/jax2tf/tests/back_compat_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -356,7 +356,7 @@ def test_tpu_Qr(self):
356356
# For lax.linalg.qr
357357
func = lambda: CompatTest.qr_harness((3, 3), np.float32)
358358
data = self.load_testdata(tpu_Qr.data_2023_03_17)
359-
self.run_one_test(func, data, rtol=1e-3)
359+
self.run_one_test(func, data, rtol=1e-3, atol=1e-3)
360360

361361
@staticmethod
362362
def lu_harness(shape, dtype):

0 commit comments

Comments
 (0)