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

Re-enable exhaustive_binary_16_bit_test_gpu.

Browse Source 
Instead disable the specific test that is failing.

PiperOrigin-RevId: 333688146
Change-Id: Ib8ffa77c66934f21e55c22f5e38b6e0d885bf9eb
This commit is contained in:
Adrian Kuegel 2020-09-25 01:25:27 -07:00 committed by TensorFlower Gardener
parent 69f5e74260
commit eaf035e953
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/compiler/xla/tests/BUILD
View File

@ -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,

2
tensorflow/compiler/xla/tests/exhaustive_binary_16_bit_test.cc
View File

@ -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;