From 48ddf64d0eb62176b903402dbcb67eae4e3e3475 Mon Sep 17 00:00:00 2001
From: Chris Leary <leary@google.com>
Date: Tue, 12 Sep 2017 21:29:34 -0700
Subject: [PATCH] [XLA] Make large params test only run in opt builds.

PiperOrigin-RevId: 168491913
---
 tensorflow/compiler/xla/tests/params_test.cc | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/tensorflow/compiler/xla/tests/params_test.cc b/tensorflow/compiler/xla/tests/params_test.cc
index 3301e4c6ee4..fda4389f479 100644
--- a/tensorflow/compiler/xla/tests/params_test.cc
+++ b/tensorflow/compiler/xla/tests/params_test.cc
@@ -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());