[XLA] Add regression test for a corner case using Reduce that currently fails with the GPU backend.

PiperOrigin-RevId: 163287986

1 files changed, 19 insertions, 0 deletions

diff --git a/tensorflow/compiler/xla/tests/reduce_test.cc b/tensorflow/compiler/xla/tests/reduce_test.cc
index b22866fc84..2589070188 100644
--- a/tensorflow/compiler/xla/tests/reduce_test.cc
+++ b/tensorflow/compiler/xla/tests/reduce_test.cc
@@ -795,6 +795,25 @@ INSTANTIATE_TEST_CASE_P(
                       BoundsLayout{{2, 300, 784}, {2, 1, 0}, {1}},
                       BoundsLayout{{2, 300, 784}, {2, 1, 0}, {0}}));
 
+// TODO(b/64093391) Disabled on GPU due to an assertion failure when running
+// IrEmitterUnnested::EmitInitializer() for the Reduce operator.  Failed on
+// 2017-07-26.
+XLA_TEST_F(ReduceTest, DISABLED_ON_GPU(OperationOnConstantAsInitValue)) {
+  ComputationBuilder builder(client_, TestName());
+  Computation max_f32 = CreateScalarMaxComputation(F32, &builder);
+
+  auto a = builder.ConstantR0<float>(2.0f);
+  auto a2 = builder.Abs(a);
+
+  std::unique_ptr<Literal> b_literal = Literal::CreateR1<float>({1.0f, 4.0f});
+  std::unique_ptr<GlobalData> b_data =
+      client_->TransferToServer(*b_literal).ConsumeValueOrDie();
+  auto b = builder.Parameter(0, b_literal->shape(), "b");
+  auto max = builder.Reduce(b, a2, max_f32, {0});
+
+  ComputeAndCompareR0<float>(&builder, 4.0f, {b_data.get()});
+}
+
 }  // namespace
 }  // namespace xla