diff --git a/tensorflow/compiler/tf2xla/BUILD b/tensorflow/compiler/tf2xla/BUILD index 0cb97d34fb4..3c9e92ceee2 100644 --- a/tensorflow/compiler/tf2xla/BUILD +++ b/tensorflow/compiler/tf2xla/BUILD @@ -5,7 +5,6 @@ load( ) load("//tensorflow/compiler/xla:xla.bzl", "xla_proto_library", "xla_py_proto_library") load("//tensorflow:tensorflow.bzl", "tf_portable_proto_library") -load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm_is_configured") package( default_visibility = [":internal"], @@ -214,8 +213,6 @@ cc_library( "xla_cpu_backend.cc", ] + if_cuda_is_configured([ "xla_gpu_backend.cc", - ]) + if_rocm_is_configured([ - "xla_gpu_backend.cc", ]), hdrs = [ "const_analysis.h", diff --git a/tensorflow/core/nccl/BUILD b/tensorflow/core/nccl/BUILD index 09f30c7679a..1931d0fa47c 100644 --- a/tensorflow/core/nccl/BUILD +++ b/tensorflow/core/nccl/BUILD @@ -52,7 +52,6 @@ tf_cuda_cc_test( srcs = ["nccl_manager_test.cc"], tags = tf_cuda_tests_tags() + [ "no_cuda_on_cpu_tap", # TODO(b/120284216): re-enable multi_gpu - "rocm_multi_gpu", # this test fails on ROCm unless 4 GPUs are used ], deps = [ "//tensorflow/core:test", diff --git a/tensorflow/python/keras/optimizer_v2/BUILD b/tensorflow/python/keras/optimizer_v2/BUILD index d6e000c82e2..633e09b70ad 100644 --- a/tensorflow/python/keras/optimizer_v2/BUILD +++ b/tensorflow/python/keras/optimizer_v2/BUILD @@ -215,7 +215,6 @@ cuda_py_test( ], shard_count = 8, tags = [ - "no_rocm", "no_windows", ], ) diff --git a/tensorflow/python/kernel_tests/linalg/linear_operator_circulant_test.py b/tensorflow/python/kernel_tests/linalg/linear_operator_circulant_test.py index e63a09c5e35..0ba456d1083 100644 --- a/tensorflow/python/kernel_tests/linalg/linear_operator_circulant_test.py +++ b/tensorflow/python/kernel_tests/linalg/linear_operator_circulant_test.py @@ -357,11 +357,6 @@ class LinearOperatorCirculantTestNonHermitianSpectrum( self.evaluate(operator.assert_non_singular()) def test_assert_non_singular_does_not_fail_for_non_singular_operator(self): - - if test.is_built_with_rocm(): - # ROCm does not yet support BLAS operations with complex types. - self.skipTest("ROCm does not support BLAS operations for complex types") - spectrum = math_ops.cast([-3j, 4 + 0j, 2j + 2], dtypes.complex64) operator = linalg.LinearOperatorCirculant(spectrum) with self.cached_session(): @@ -661,11 +656,6 @@ class LinearOperatorCirculant3DTest(test.TestCase): yield sess def test_real_spectrum_gives_self_adjoint_operator(self): - - if test.is_built_with_rocm(): - # ROCm does not yet support BLAS operations with complext types - self.skipTest("ROCm does not support BLAS operations for complex types") - with self.cached_session(): # This is a real and hermitian spectrum. spectrum = linear_operator_test_util.random_normal( @@ -682,11 +672,6 @@ class LinearOperatorCirculant3DTest(test.TestCase): self.assertAllClose(matrix, matrix_h) def test_defining_operator_using_real_convolution_kernel(self): - - if test.is_built_with_rocm(): - # ROCm does not yet support BLAS operations with complext types - self.skipTest("ROCm does not support BLAS operations for complex types") - with self.cached_session(): convolution_kernel = linear_operator_test_util.random_normal( shape=(2, 2, 3, 5), dtype=dtypes.float32) @@ -705,11 +690,6 @@ class LinearOperatorCirculant3DTest(test.TestCase): np.testing.assert_allclose(0, np.imag(matrix), atol=1e-5) def test_defining_spd_operator_by_taking_real_part(self): - - if test.is_built_with_rocm(): - # ROCm does not yet support BLAS operations with complext types - self.skipTest("ROCm does not support BLAS operations for complex types") - with self.cached_session(): # Necessary for fft_kernel_label_map # S is real and positive. s = linear_operator_test_util.random_uniform( diff --git a/tensorflow/tools/ci_build/linux/rocm/run_csb_tests.sh b/tensorflow/tools/ci_build/linux/rocm/run_csb_tests.sh index 401d2e748f7..3eca1f53767 100755 --- a/tensorflow/tools/ci_build/linux/rocm/run_csb_tests.sh +++ b/tensorflow/tools/ci_build/linux/rocm/run_csb_tests.sh @@ -38,7 +38,7 @@ yes "" | $PYTHON_BIN_PATH configure.py bazel test \ --config=rocm \ -k \ - --test_tag_filters=gpu,-no_gpu,-no_rocm,-benchmark-test,-no_oss,-oss_serial,-rocm_multi_gpu, \ + --test_tag_filters=gpu,-no_gpu,-no_rocm,-benchmark-test,-no_oss,-oss_serial, \ --test_timeout 600,900,2400,7200 \ --test_output=errors \ --jobs=${N_JOBS} \ @@ -50,18 +50,4 @@ bazel test \ -//tensorflow/compiler/... \ -//tensorflow/lite/... \ -//tensorflow/python/compiler/tensorrt/... \ -&& bazel test \ - --config=rocm \ - -k \ - --test_tag_filters=-no_rocm,rocm_multi_gpu, \ - --test_timeout 600,900,2400,7200 \ - --test_output=errors \ - --jobs=${N_JOBS} \ - --local_test_jobs=1 \ - --test_sharding_strategy=disabled \ - -- \ - //tensorflow/... \ - -//tensorflow/compiler/... \ - -//tensorflow/lite/... \ - -//tensorflow/python/compiler/tensorrt/... \