diff --git a/test/test_renderer_failures.py b/test/test_renderer_failures.py index e428271f7f..2db5b41389 100644 --- a/test/test_renderer_failures.py +++ b/test/test_renderer_failures.py @@ -85,7 +85,7 @@ class TestPTXFailures(unittest.TestCase): b = Tensor.randn(34, 32, dtype=dtypes.half).realize() result = a.pad((1,1)).matmul(b, dtype=dtypes.half).numpy() reference = a.pad((1,1)).matmul(b, dtype=dtypes.float).numpy() - np.testing.assert_allclose(result, reference) + np.testing.assert_allclose(result, reference, atol=1e-2, rtol=1e-2) if __name__ == '__main__': unittest.main()