From b76344d0ab78f45f32abb9c32127f8be6b426f7f Mon Sep 17 00:00:00 2001
From: David Majnemer <majnemer@google.com>
Date: Tue, 10 Jan 2017 13:53:29 -0800
Subject: [PATCH] Add an XLA test for ComputationBuilder::SqrtF32(0) Change:
 144124103

---
 .../compiler/xla/tests/scalar_computations_test.cc | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/tensorflow/compiler/xla/tests/scalar_computations_test.cc b/tensorflow/compiler/xla/tests/scalar_computations_test.cc
index bd9cae4d1dd..d618c883d83 100644
--- a/tensorflow/compiler/xla/tests/scalar_computations_test.cc
+++ b/tensorflow/compiler/xla/tests/scalar_computations_test.cc
@@ -608,6 +608,20 @@ TEST_F(ScalarComputationsTest, ComplicatedArithmeticExpressionS32) {
   ComputeAndCompareR0<int32>(&b, 10, {});
 }
 
+TEST_F(ScalarComputationsTest, SqrtF320) {
+  ComputationBuilder builder(client_, TestName());
+  Literal zero_literal = LiteralUtil::Zero(PrimitiveType::F32);
+
+  std::unique_ptr<GlobalData> zero_data =
+      client_->TransferToServer(zero_literal).ConsumeValueOrDie();
+
+  ComputationDataHandle zero =
+      builder.Parameter(0, zero_literal.shape(), "zero");
+  builder.SqrtF32(zero);
+
+  ComputeAndCompareR0<float>(&builder, 0.0f, {zero_data.get()}, error_spec_);
+}
+
 }  // namespace
 }  // namespace xla