From 88be89ea85bf82b1a28f9dabe223a4690563a359 Mon Sep 17 00:00:00 2001 From: Christian Sigg Date: Wed, 1 Jul 2020 12:39:50 -0700 Subject: [PATCH] Disable tests for CUDA 11 that fail CI. PiperOrigin-RevId: 319277399 Change-Id: I437121a8b44ea1da717eb2452a768103908c51ca --- tensorflow/python/keras/distribute/BUILD | 1 + tensorflow/python/keras/layers/BUILD | 3 +++ tensorflow/python/keras/tests/BUILD | 1 + 3 files changed, 5 insertions(+) diff --git a/tensorflow/python/keras/distribute/BUILD b/tensorflow/python/keras/distribute/BUILD index 9fa2f315720..3935cbb59f2 100644 --- a/tensorflow/python/keras/distribute/BUILD +++ b/tensorflow/python/keras/distribute/BUILD @@ -120,6 +120,7 @@ distribute_py_test( main = "custom_training_loop_models_test.py", tags = [ "multi_and_single_gpu", + "no_cuda11", ], tpu_tags = [ "no_oss", # b/153615544. diff --git a/tensorflow/python/keras/layers/BUILD b/tensorflow/python/keras/layers/BUILD index f67838d591a..b9f00f65d74 100644 --- a/tensorflow/python/keras/layers/BUILD +++ b/tensorflow/python/keras/layers/BUILD @@ -519,6 +519,7 @@ cuda_py_test( python_version = "PY3", shard_count = 4, tags = [ + "no_cuda11", "no_windows_gpu", ], deps = [ @@ -784,6 +785,7 @@ cuda_py_test( srcs = ["lstm_v2_test.py"], python_version = "PY3", shard_count = 12, + tags = ["no_cuda11"], deps = [ "//tensorflow/python:client_testlib", "//tensorflow/python/keras", @@ -798,6 +800,7 @@ cuda_py_test( srcs = ["gru_v2_test.py"], python_version = "PY3", shard_count = 12, + tags = ["no_cuda11"], deps = [ "//tensorflow/python:client_testlib", "//tensorflow/python/keras", diff --git a/tensorflow/python/keras/tests/BUILD b/tensorflow/python/keras/tests/BUILD index 88273f4bb2b..35f35252ebd 100644 --- a/tensorflow/python/keras/tests/BUILD +++ b/tensorflow/python/keras/tests/BUILD @@ -274,6 +274,7 @@ cuda_py_test( name = "op_callbacks_test", srcs = ["op_callbacks_test.py"], python_version = "PY3", + tags = ["no_cuda11"], deps = [ "//tensorflow/python:framework_ops", "//tensorflow/python:framework_test_lib",