experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

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

Browse Source 
PiperOrigin-RevId: 163287986
This commit is contained in:
A. Unique TensorFlower 2017-07-26 18:42:24 -07:00 committed by TensorFlower Gardener
parent 32e198f2d5
commit fd5de4690d
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tensorflow/compiler/xla/tests/reduce_test.cc
View File

@ -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