From eaf035e953e06e5564bd197c6c3105b7fe5130e4 Mon Sep 17 00:00:00 2001
From: Adrian Kuegel <akuegel@google.com>
Date: Fri, 25 Sep 2020 01:25:27 -0700
Subject: [PATCH] Re-enable exhaustive_binary_16_bit_test_gpu.

Instead disable the specific test that is failing.

PiperOrigin-RevId: 333688146
Change-Id: Ib8ffa77c66934f21e55c22f5e38b6e0d885bf9eb
---
 tensorflow/compiler/xla/tests/BUILD                            | 3 +--
 tensorflow/compiler/xla/tests/exhaustive_binary_16_bit_test.cc | 2 +-
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/tensorflow/compiler/xla/tests/BUILD b/tensorflow/compiler/xla/tests/BUILD
index e9437cf329b..19a7b69b98c 100644
--- a/tensorflow/compiler/xla/tests/BUILD
+++ b/tensorflow/compiler/xla/tests/BUILD
@@ -858,8 +858,7 @@ xla_test(
     name = "exhaustive_binary_16_bit_test",
     srcs = ["exhaustive_binary_16_bit_test.cc"],
     backends = [
-        # TODO (b/169384515) enable gpu after bug is fixed
-        # "gpu",
+        "gpu",
         "cpu",
     ],
     shard_count = 48,
diff --git a/tensorflow/compiler/xla/tests/exhaustive_binary_16_bit_test.cc b/tensorflow/compiler/xla/tests/exhaustive_binary_16_bit_test.cc
index dca8e31e792..f884bb9c0e0 100644
--- a/tensorflow/compiler/xla/tests/exhaustive_binary_16_bit_test.cc
+++ b/tensorflow/compiler/xla/tests/exhaustive_binary_16_bit_test.cc
@@ -123,7 +123,7 @@ BINARY_TEST_16BIT(Min, {
 })
 
 // TODO(bixia): Pow fails with bfloat16 on CPU.
-BINARY_TEST_16BIT(DISABLED_ON_CPU(Pow), {
+BINARY_TEST_16BIT(DISABLED_ON_GPU(DISABLED_ON_CPU(Pow)), {
   // See b/162664705.
   known_incorrect_fn_ = [](int64 val) {
     Eigen::bfloat16 f;