diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index 94b11feb823..469dfbca6f3 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -3669,6 +3669,7 @@ cuda_py_test( "//third_party/py/numpy", ], shard_count = 16, + tags = ["no_rocm"], ) cuda_py_test( diff --git a/tensorflow/python/eager/BUILD b/tensorflow/python/eager/BUILD index 47a7e9ae2f5..a6a367e2685 100644 --- a/tensorflow/python/eager/BUILD +++ b/tensorflow/python/eager/BUILD @@ -151,6 +151,7 @@ cuda_py_test( "//tensorflow/python:nn_grad", "//tensorflow/python:training", ], + tags = ["no_rocm"], ) cuda_py_test( @@ -530,6 +531,7 @@ tf_xla_py_test( srcs = ["def_function_xla_test.py"], tags = [ "no_pip", + "no_rocm", "nomac", ], deps = [ diff --git a/tensorflow/python/keras/BUILD b/tensorflow/python/keras/BUILD index 9e60d0f1002..82a526a6458 100755 --- a/tensorflow/python/keras/BUILD +++ b/tensorflow/python/keras/BUILD @@ -612,7 +612,10 @@ cuda_py_test( "//tensorflow/python:client_testlib", ], shard_count = 4, - tags = ["no_windows_gpu"], + tags = [ + "no_rocm", + "no_windows_gpu", + ], ) tf_py_test( @@ -783,6 +786,7 @@ cuda_py_test( "//tensorflow/python:client_testlib", ], shard_count = 8, + tags = ["no_rocm"], ) cuda_py_test( @@ -796,6 +800,7 @@ cuda_py_test( "//tensorflow/python:client_testlib", ], shard_count = 8, + tags = ["no_rocm"], ) tf_py_test( @@ -1186,6 +1191,7 @@ tf_py_test( "//third_party/py/numpy", "//tensorflow/python:client_testlib", ], + tags = ["no_rocm"], ) tf_py_test( diff --git a/tensorflow/python/kernel_tests/BUILD b/tensorflow/python/kernel_tests/BUILD index 86c63496fa8..1274c82b4bd 100644 --- a/tensorflow/python/kernel_tests/BUILD +++ b/tensorflow/python/kernel_tests/BUILD @@ -131,6 +131,7 @@ cuda_py_test( "//tensorflow/python:client_testlib", ], grpc_enabled = True, + tags = ["no_rocm"], xla_enable_strict_auto_jit = True, ) @@ -1688,6 +1689,7 @@ cuda_py_test( "//tensorflow/python:framework_for_generated_wrappers", "//tensorflow/python:nn_ops", ], + tags = ["no_rocm"], xla_enable_strict_auto_jit = True, ) @@ -2605,6 +2607,7 @@ cuda_py_test( "//tensorflow/python/eager:context", ], flaky = 1, # create_local_cluster sometimes times out. + tags = ["no_rocm"], xla_enable_strict_auto_jit = True, ) @@ -2764,6 +2767,7 @@ cuda_py_test( ], shard_count = 2, tags = [ + "no_rocm", "optonly", # flaky timeouts unless optimized ], xla_enable_strict_auto_jit = True, @@ -2805,6 +2809,7 @@ cuda_py_test( ], shard_count = 4, tags = [ + "no_rocm", "optonly", # times out ], xla_enable_strict_auto_jit = True, @@ -2868,6 +2873,7 @@ cuda_py_test( "//tensorflow/python:nn_grad", "//tensorflow/python:nn_ops", ], + tags = ["no_rocm"], xla_enable_strict_auto_jit = True, ) @@ -2887,6 +2893,7 @@ cuda_py_test( "//tensorflow/python:nn_ops_gen", ], shard_count = 4, + tags = ["no_rocm"], xla_enable_strict_auto_jit = True, ) @@ -3147,6 +3154,7 @@ cuda_py_test( "//tensorflow/python:nn_ops", ], shard_count = 30, + tags = ["no_rocm"], xla_enable_strict_auto_jit = True, ) @@ -3285,7 +3293,10 @@ cuda_py_test( ], data = ["//tensorflow/python/kernel_tests/testdata:self_adjoint_eig_op_test_files"], shard_count = 20, - tags = ["no_windows"], + tags = [ + "no_rocm", # flaky test + "no_windows", + ], # TODO(kuny): Add xla_enable_strict_auto_jit = True after b/124377352 is fixed. ) diff --git a/tensorflow/python/kernel_tests/signal/BUILD b/tensorflow/python/kernel_tests/signal/BUILD index 554bf380294..4caecc85ca5 100644 --- a/tensorflow/python/kernel_tests/signal/BUILD +++ b/tensorflow/python/kernel_tests/signal/BUILD @@ -29,6 +29,7 @@ cuda_py_tests( "//tensorflow/python:spectral_ops_test_util", "//tensorflow/python/ops/signal", ], + tags = ["no_rocm"], xla_enable_strict_auto_jit = True, ) @@ -45,7 +46,10 @@ cuda_py_tests( "//tensorflow/python/ops/signal", ], shard_count = 4, - tags = ["optonly"], + tags = [ + "no_rocm", + "optonly", + ], xla_enable_strict_auto_jit = True, ) @@ -73,6 +77,7 @@ cuda_py_tests( "//tensorflow/python/ops/signal", "//tensorflow/python:spectral_ops_test_util", ], + tags = ["no_rocm"], xla_enable_strict_auto_jit = True, ) @@ -130,7 +135,10 @@ cuda_py_tests( "//tensorflow/python:spectral_ops_test_util", "//tensorflow/python/ops/signal", ], - tags = ["nomac"], + tags = [ + "no_rocm", + "nomac", + ], xla_enable_strict_auto_jit = True, ) diff --git a/tensorflow/python/ops/parallel_for/BUILD b/tensorflow/python/ops/parallel_for/BUILD index b1e12b24b9a..0a2f3e25a29 100644 --- a/tensorflow/python/ops/parallel_for/BUILD +++ b/tensorflow/python/ops/parallel_for/BUILD @@ -115,6 +115,7 @@ cuda_py_test( "//tensorflow/python:random_ops", "//tensorflow/python:util", ], + tags = ["no_rocm"], ) cuda_py_test( diff --git a/tensorflow/python/training/checkpointable/BUILD b/tensorflow/python/training/checkpointable/BUILD index a7ddd812583..40a6e93e09c 100644 --- a/tensorflow/python/training/checkpointable/BUILD +++ b/tensorflow/python/training/checkpointable/BUILD @@ -195,6 +195,7 @@ tf_xla_py_test( srcs = ["util_xla_test.py"], tags = [ "no_pip", + "no_rocm", "nomac", "notsan", # b/74395663 ], diff --git a/tensorflow/tools/api/tests/BUILD b/tensorflow/tools/api/tests/BUILD index 1675dfa512d..b81ce692bbd 100644 --- a/tensorflow/tools/api/tests/BUILD +++ b/tensorflow/tools/api/tests/BUILD @@ -23,6 +23,7 @@ py_test( "//tensorflow/tools/api/tests:README.txt", ], srcs_version = "PY2AND3", + tags = ["no_rocm"], deps = [ "//tensorflow:tensorflow_py", "//tensorflow/python:client_testlib",