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;