diff --git a/tensorflow/compiler/xla/tests/reduce_test.cc b/tensorflow/compiler/xla/tests/reduce_test.cc index b22866fc84b..25890701880 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(2.0f); + auto a2 = builder.Abs(a); + + std::unique_ptr b_literal = Literal::CreateR1({1.0f, 4.0f}); + std::unique_ptr 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(&builder, 4.0f, {b_data.get()}); +} + } // namespace } // namespace xla