From 42a64ba0d9e0973b1707fcd9b8bd8d14b2d4e3e5 Mon Sep 17 00:00:00 2001 From: ursk Date: Thu, 31 Aug 2023 13:30:00 -0700 Subject: [PATCH] Update test tolerances. PiperOrigin-RevId: 561744940 --- tensorflow_probability/python/internal/backend/numpy/BUILD | 4 +++- .../python/internal/backend/numpy/numpy_test.py | 4 ++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/tensorflow_probability/python/internal/backend/numpy/BUILD b/tensorflow_probability/python/internal/backend/numpy/BUILD index 9d007f7ac3..48f890c4a0 100644 --- a/tensorflow_probability/python/internal/backend/numpy/BUILD +++ b/tensorflow_probability/python/internal/backend/numpy/BUILD @@ -479,7 +479,9 @@ py_test( "--test_mode=xla", # TODO(b/168718272): reduce_*([nan, nan], axis=0) (GPU) # histogram_fixed_width_bins fails with f32([0.]), [0.0, 0.0], 2 - "--xla_disabled=math.cumulative_logsumexp,math.reduce_min,math.reduce_max,histogram_fixed_width_bins", + ("--xla_disabled=math.cumulative_logsumexp,math.reduce_min,math.reduce_max,histogram_fixed_width_bins," + + # TODO(b/298426124): TF floomod GPU bug + "math.floormod"), ], main = "numpy_test.py", shard_count = 11, diff --git a/tensorflow_probability/python/internal/backend/numpy/numpy_test.py b/tensorflow_probability/python/internal/backend/numpy/numpy_test.py index 5cea3d1aa4..4f6b75be83 100644 --- a/tensorflow_probability/python/internal/backend/numpy/numpy_test.py +++ b/tensorflow_probability/python/internal/backend/numpy/numpy_test.py @@ -1377,10 +1377,10 @@ def _not_implemented(*args, **kwargs): [n_same_shape(n=2, elements=[floats(), non_zero_floats()])]), TestCase('math.xlogy', [n_same_shape(n=2, elements=[floats(), positive_floats()])], - atol=1e-4), + atol=1e-4, rtol=1e-3), TestCase('math.xlog1py', [n_same_shape(n=2, elements=[floats(), positive_floats()])], - atol=1e-4), + atol=1e-4, rtol=1e-3), TestCase('nn.conv2d', [conv2d_params()], disabled=NUMPY_MODE), TestCase( 'nn.sparse_softmax_cross_entropy_with_logits', [sparse_xent_params()],