diff --git a/tensorflow/compiler/xla/tests/prng_test.cc b/tensorflow/compiler/xla/tests/prng_test.cc index 201c0da87f1..1a95f2fb549 100644 --- a/tensorflow/compiler/xla/tests/prng_test.cc +++ b/tensorflow/compiler/xla/tests/prng_test.cc @@ -122,7 +122,7 @@ XLA_TEST_F(PrngTest, DISABLED_ON_INTERPRETER(DISABLED_ON_GPU( bfloat16 interval = static_cast(0.25); std::vector counts(static_cast((high - low) / interval), 0); - constexpr int64 count = 100; + constexpr int64 count = 1000; for (int64 seed = 0; seed < count; ++seed) { auto result = UniformTest(low, high, {}, /*seed=*/seed); result.EachCell([&](absl::Span, bfloat16 value) {