diff --git a/tensorflow/python/kernel_tests/BUILD b/tensorflow/python/kernel_tests/BUILD index 5eabd8e53fd..179298c837e 100644 --- a/tensorflow/python/kernel_tests/BUILD +++ b/tensorflow/python/kernel_tests/BUILD @@ -2916,8 +2916,7 @@ cuda_py_test( "no_rocm", "optonly", # flaky timeouts unless optimized ], - # TODO(b/134677329): Re-enable this test. - xla_enable_strict_auto_jit = False, + xla_enable_strict_auto_jit = True, ) cuda_py_test( diff --git a/tensorflow/python/kernel_tests/conv2d_backprop_filter_grad_test.py b/tensorflow/python/kernel_tests/conv2d_backprop_filter_grad_test.py index a2feb67dc73..c6e5945bd75 100644 --- a/tensorflow/python/kernel_tests/conv2d_backprop_filter_grad_test.py +++ b/tensorflow/python/kernel_tests/conv2d_backprop_filter_grad_test.py @@ -110,7 +110,7 @@ class Conv2DBackpropFilterGradTest(test.TestCase): [in_val, out_backprop_val], [in_shape, out_backprop_shape], output, filter_shape) print("conv2d_backprop_filter gradient err = %g " % err) - err_tolerance = 2e-3 + err_tolerance = 1e-2 self.assertLess(err, err_tolerance)