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

[XLA] Make large params test only run in opt builds.

Browse Source 
PiperOrigin-RevId: 168491913
This commit is contained in:
Chris Leary 2017-09-12 21:29:34 -07:00 committed by TensorFlower Gardener
parent 11d3ac29d5
commit 48ddf64d0e
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/compiler/xla/tests/params_test.cc
View File

@ -251,6 +251,10 @@ XLA_TEST_F(ParamsTest, HundredLargeR1Parameters) {
ComputeAndCompareR1<float>(&builder, sum, param_data, ErrorSpec(0.0001f));
}
// Only run the 3,000-parameter tests in opt mode to avoid test timeouts.
// Timeout last observed on 2017-09-12.
#ifndef NDEBUG
// TODO(b/65525254) Fails on GPU on 2017-09-10 because we try to reserve too
// much space in parameter memory for the kernel.
//
@ -330,6 +334,8 @@ XLA_TEST_F(ParamsTest, DISABLED_ON_CPU(DISABLED_ON_GPU(
ComputeAndCompareTuple(&builder, *Literal::MakeTuple(ptrs), param_data);
}
#endif
XLA_TEST_F(ParamsTest,
DISABLED_ON_CPU_PARALLEL(TupleOfR1ParametersAddedTogether)) {
ComputationBuilder builder(client_, TestName());