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

Merge pull request from ROCmSoftwarePlatform:google_upstream_rocm_add_remove_no_rocm_tag

Browse Source 
PiperOrigin-RevId: 348425014
Change-Id: I1f4ec5ab94f27ea201a9dbc8b53d337d03e01d6c
This commit is contained in:
TensorFlower Gardener 2020-12-21 00:32:48 -08:00
commit ee11cf13ea
25 changed files with 23 additions and 43 deletions
tensorflow
compiler
tests
BUILD
xla
service
gpu/tests
BUILD
mlir_gpu/experimental/conv_emitter
BUILD
tests
BUILD
core
common_runtime
BUILD
nccl
BUILD
python
BUILD
compiler/tensorrt
BUILD
debug
BUILD
distribute
BUILD
keras
BUILD
distribute
BUILD
engine
BUILD
layers
BUILD
tests
BUILD
kernel_tests
BUILD
distributions
BUILD
linalg
BUILD
sparse
BUILD
signal
BUILD
ops/ragged
BUILD
saved_model
BUILD
tools
BUILD
tpu
BUILD
tools/compatibility
BUILD

5
tensorflow/compiler/tests/BUILD
View File

@ -477,7 +477,6 @@ tf_xla_py_test(
python_version = "PY3",
tags = [
"no_pip", # TODO(b/149738646): fix pip install so these tests run on kokoro pip
"no_rocm",
],
deps = [
":xla_test",
@ -1629,7 +1628,6 @@ cuda_py_test(
shard_count = 5,
tags = [
"no_pip", # TODO(b/149738646): fix pip install so these tests run on kokoro pip
"no_rocm",
],
xla_enable_strict_auto_jit = False,
xla_enabled = True,
@ -1654,7 +1652,6 @@ cuda_py_test(
srcs = ["dense_layer_test.py"],
tags = [
"no_pip", # TODO(b/149738646): fix pip install so these tests run on kokoro pip
"no_rocm",
],
xla_enable_strict_auto_jit = False,
xla_enabled = True,
@ -1749,7 +1746,6 @@ cuda_py_test(
srcs = ["lstm_test.py"],
tags = [
"no_pip", # TODO(b/149738646): fix pip install so these tests run on kokoro pip
"no_rocm",
],
xla_enable_strict_auto_jit = False,
xla_enabled = True,
@ -1873,7 +1869,6 @@ tf_xla_py_test(
tags = [
"no_oss", # TODO(b/148108508): Re-enable this test in OSS.
"no_pip", # TODO(b/149738646): fix pip install so these tests run on kokoro pip
"no_rocm",
],
deps = [
":xla_test",

6
tensorflow/compiler/xla/service/gpu/tests/BUILD
View File

@ -172,7 +172,7 @@ tf_cc_test(
srcs = [
"reduction_vectorization_test.cc",
],
tags = tf_cuda_tests_tags() + ["no_rocm"],
tags = tf_cuda_tests_tags(),
deps = [
":gpu_codegen_test",
"//tensorflow/compiler/xla:debug_options_flags",
@ -410,7 +410,7 @@ tf_cc_test(
tf_cc_test(
name = "gpu_unrolling_test",
srcs = ["gpu_unrolling_test.cc"],
tags = tf_cuda_tests_tags(),
tags = tf_cuda_tests_tags() + ["no_rocm"],
deps = [
":gpu_codegen_test",
"//tensorflow/compiler/xla/service:hlo_module_config",
@ -441,7 +441,7 @@ tf_cc_test(
tf_cc_test(
name = "gpu_atomic_test",
srcs = ["gpu_atomic_test.cc"],
tags = tf_cuda_tests_tags(),
tags = tf_cuda_tests_tags() + ["no_rocm"],
deps = [
":gpu_codegen_test",
"//tensorflow/compiler/xla/tests:filecheck",

1
tensorflow/compiler/xla/service/mlir_gpu/experimental/conv_emitter/BUILD
View File

@ -63,7 +63,6 @@ tf_cc_test(
srcs = ["conv_emitter_test.cc"],
tags = [
"no_oss", # TODO(b/148143101): Test should pass in OSS.
"no_rocm",
],
deps = [
":conv_emitter",

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

@ -367,6 +367,7 @@ xla_test(
"conv_depthwise_test.cc",
],
shard_count = 50,
tags = ["no_rocm"], # ROCm 3.9 regression
deps = [
":conv_depthwise_common",
":test_macros_header",
@ -388,6 +389,7 @@ xla_test(
timeout = "long",
srcs = ["conv_depthwise_backprop_filter_test.cc"],
shard_count = 40,
tags = ["no_rocm"], # ROCm 3.9 regression
deps = [
":test_macros_header",
"//tensorflow/compiler/xla:execution_options_util",
@ -412,6 +414,7 @@ xla_test(
"cpu",
],
shard_count = 50,
tags = ["no_rocm"], # ROCm 3.9 regression
deps = [
":client_library_test_base",
":hlo_test_base",
@ -626,7 +629,6 @@ xla_test(
name = "conditional_test",
srcs = ["conditional_test.cc"],
shard_count = 2,
tags = ["no_rocm"],
deps = [
":test_macros_header",
"//tensorflow/compiler/xla:xla_data_proto_cc",
@ -665,7 +667,6 @@ xla_test(
name = "scalar_computations_test",
srcs = ["scalar_computations_test.cc"],
shard_count = 32,
tags = ["no_rocm"],
deps = [
":test_macros_header",
"//tensorflow/compiler/xla:literal",
@ -923,7 +924,7 @@ xla_test(
srcs = ["dot_operation_test.cc"],
shard_count = 20,
tags = [
"no_rocm",
"no_rocm", # ROCm 3.9 regression
"optonly",
],
deps = [
@ -957,7 +958,7 @@ xla_test(
backends = ["gpu"],
shard_count = 20,
tags = [
"no_rocm",
"no_rocm", # ROCm 3.9 regression
"optonly",
# TODO(b/151340488): Timed out on 2020-03-12.
"nozapfhahn",
@ -1024,7 +1025,7 @@ xla_test(
},
shard_count = 20,
tags = [
"no_rocm",
"no_rocm", # ROCm 3.9 regression
"optonly",
],
deps = [
@ -1252,6 +1253,7 @@ xla_test(
"cpu": ["nomsan"],
},
shard_count = 30,
tags = ["no_rocm"], # ROCm 3.9 regression
deps = [
":test_macros_header",
"//tensorflow/compiler/xla:array3d",
@ -1276,6 +1278,7 @@ xla_test(
timeout = "long",
srcs = ["convolution_dimension_numbers_test.cc"],
shard_count = 20,
tags = ["no_rocm"], # ROCm 3.9 regression
deps = [
":test_macros_header",
"//tensorflow/compiler/xla:array4d",
@ -1514,7 +1517,6 @@ xla_test(
srcs = ["reduce_test.cc"],
shard_count = 31,
tags = [
"no_rocm",
"optonly",
],
deps = [
@ -1594,7 +1596,6 @@ xla_test(
timeout = "long",
srcs = ["select_and_scatter_test.cc"],
tags = [
"no_rocm",
"nozapfhahn",
"optonly",
],
@ -2321,6 +2322,7 @@ xla_test(
name = "multioutput_fusion_test",
srcs = ["multioutput_fusion_test.cc"],
backends = ["gpu"],
tags = ["no_rocm"], # ROCm 3.9 regression
deps = [
":test_macros_header",
"//tensorflow/compiler/xla:literal",

1
tensorflow/core/common_runtime/BUILD
View File

@ -2127,7 +2127,6 @@ tf_cuda_cc_test(
size = "small",
srcs = ["process_function_library_runtime_test.cc"],
linkstatic = tf_kernel_tests_linkstatic(),
tags = ["no_rocm"],
deps = [
":core_cpu",
":core_cpu_internal",

2
tensorflow/core/nccl/BUILD
View File

@ -64,8 +64,6 @@ tf_cuda_cc_test(
"manual",
"multi_gpu",
"no_oss",
# TODO(b/147451637): Replace 'no_rocm' with 'rocm_multi_gpu'.
"no_rocm",
"notap",
],
deps = [

3
tensorflow/python/BUILD
View File

@ -2018,6 +2018,7 @@ cuda_py_test(
python_version = "PY3",
shard_count = 10,
tags = [
"no_rocm",
"noasan",
"optonly",
],
@ -2063,6 +2064,7 @@ tf_py_test(
srcs = ["framework/importer_test.py"],
main = "framework/importer_test.py",
python_version = "PY3",
tags = ["no_rocm"],
deps = [
":array_ops",
":client_testlib",
@ -4883,6 +4885,7 @@ cuda_py_test(
srcs = ["ops/nn_fused_batchnorm_test.py"],
python_version = "PY3",
shard_count = 24,
tags = ["no_rocm"],
deps = [
":array_ops",
":client_testlib",

1
tensorflow/python/compiler/tensorrt/BUILD
View File

@ -97,7 +97,6 @@ cuda_py_test(
tags = [
"no_cuda_on_cpu_tap",
"no_pip",
"no_rocm",
"no_windows",
"nomac",
],

1
tensorflow/python/debug/BUILD
View File

@ -1119,7 +1119,6 @@ py_test(
srcs = ["cli/debugger_cli_common_test.py"],
python_version = "PY3",
srcs_version = "PY2AND3",
tags = ["no_rocm"],
deps = [
":debugger_cli_common",
"//tensorflow/python:framework_test_lib",

1
tensorflow/python/distribute/BUILD
View File

@ -162,7 +162,6 @@ py_test(
srcs = ["distribute_lib_test.py"],
python_version = "PY3",
srcs_version = "PY2AND3",
tags = ["no_rocm"],
deps = [
":combinations",
":distribute_lib",

1
tensorflow/python/keras/BUILD
View File

@ -582,7 +582,6 @@ tf_py_test(
python_version = "PY3",
shard_count = 8,
tags = [
"no_rocm",
"notsan", # b/67509773
],
deps = [

2
tensorflow/python/keras/distribute/BUILD
View File

@ -808,7 +808,6 @@ distribute_py_test(
tags = [
"multi_and_single_gpu",
"no_cuda_asan", # times out
"no_rocm",
],
xla_tags = [
"no_cuda_asan", # times out
@ -828,7 +827,6 @@ distribute_py_test(
shard_count = 7,
tags = [
"multi_and_single_gpu",
"no_rocm",
],
xla_tags = [
"no_cuda_asan", # times out

4
tensorflow/python/keras/engine/BUILD
View File

@ -394,7 +394,6 @@ tf_py_test(
shard_count = 20,
tags = [
"manual",
"no_rocm",
"nomac", # TODO(mihaimaruseac): b/127695564
"notsan",
],
@ -519,7 +518,6 @@ tf_py_test(
python_version = "PY3",
shard_count = 30,
tags = [
"no_rocm",
"nomac", # TODO(mihaimaruseac): b/127695564
],
deps = [
@ -609,6 +607,7 @@ tf_py_test(
shard_count = 8,
tags = [
"no-internal-py3",
"no_rocm",
"nomac", # TODO(mihaimaruseac): b/127695564
],
deps = [
@ -663,7 +662,6 @@ tf_py_test(
python_version = "PY3",
shard_count = 8,
tags = [
"no_rocm",
"nomac", # TODO(mihaimaruseac): b/127695564
],
deps = [

2
tensorflow/python/keras/layers/BUILD
View File

@ -750,7 +750,6 @@ tf_py_test(
python_version = "PY3",
shard_count = 4,
tags = [
"no_rocm",
"notsan", # http://b/62136390
],
deps = [
@ -769,7 +768,6 @@ tf_py_test(
python_version = "PY3",
shard_count = 4,
tags = [
"no_rocm",
"noasan", # times out b/63678675
"notsan", # http://b/62189182
],

1
tensorflow/python/keras/tests/BUILD
View File

@ -147,7 +147,6 @@ tf_py_test(
python_version = "PY3",
shard_count = 16,
tags = [
"no_rocm",
"notsan",
],
deps = [

3
tensorflow/python/kernel_tests/BUILD
View File

@ -241,7 +241,6 @@ cuda_py_test(
srcs = ["cholesky_op_test.py"],
shard_count = 5,
tags = [
"no_rocm", # TODO(rocm): feature not supported on ROCm platform
"nomsan", # TODO(b/131773093): Re-enable.
],
deps = [
@ -1720,6 +1719,7 @@ cuda_py_test(
name = "betainc_op_test",
size = "small",
srcs = ["betainc_op_test.py"],
tags = ["no_rocm"], # ROCm 3.9 regression
xla_tags = [
"no_cuda_asan", # times out
],
@ -3851,7 +3851,6 @@ cuda_py_test(
size = "medium",
srcs = ["tridiagonal_matmul_op_test.py"],
shard_count = 10,
tags = ["no_rocm"],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",

1
tensorflow/python/kernel_tests/distributions/BUILD
View File

@ -61,6 +61,7 @@ cuda_py_test(
size = "small",
srcs = ["beta_test.py"],
tags = [
"no_rocm", # ROCm 3.9 regression
"notsan", # b/173653918
],
xla_tags = [

1
tensorflow/python/kernel_tests/linalg/BUILD
View File

@ -149,7 +149,6 @@ cuda_py_test(
srcs = ["linear_operator_circulant_test.py"],
shard_count = 10,
tags = [
"no_rocm", # calls BLAS ops for complex types
"noasan", # times out, b/63678675
"optonly", # times out, b/79171797
],

5
tensorflow/python/kernel_tests/linalg/sparse/BUILD
View File

@ -40,7 +40,10 @@ cuda_py_test(
srcs = ["csr_sparse_matrix_ops_test.py"],
main = "csr_sparse_matrix_ops_test.py",
shard_count = 10,
tags = ["notsan"], # b/149115441
tags = [
"no_rocm", # ROCm 3.8 regression
"notsan", # b/149115441
],
deps = [
"//tensorflow/python/ops/linalg/sparse",
"//tensorflow/python/ops/linalg/sparse:gen_sparse_csr_matrix_ops",

1
tensorflow/python/kernel_tests/signal/BUILD
View File

@ -125,6 +125,7 @@ cuda_py_tests(
srcs = ["spectral_ops_test.py"],
python_version = "PY3",
tags = [
"no_rocm",
"nomac",
],
deps = [

1
tensorflow/python/ops/ragged/BUILD
View File

@ -1104,7 +1104,6 @@ py_test(
srcs = ["ragged_map_fn_op_test.py"],
python_version = "PY3",
srcs_version = "PY2AND3",
tags = ["no_rocm"],
deps = [
":ragged", # fixdeps: keep
":ragged_factory_ops",

1
tensorflow/python/saved_model/BUILD
View File

@ -367,7 +367,6 @@ py_strict_library(
tf_py_test(
name = "save_test",
srcs = ["save_test.py"],
tags = ["no_rocm"],
deps = [
":loader",
":save",

4
tensorflow/python/tools/BUILD
View File

@ -444,7 +444,6 @@ saved_model_compile_aot(
"//tensorflow/cc/saved_model:saved_model_half_plus_two",
],
force_without_xla_support_flag = False,
tags = ["no_rocm"],
)
saved_model_compile_aot(
@ -455,7 +454,6 @@ saved_model_compile_aot(
"//tensorflow/cc/saved_model:saved_model_half_plus_two",
],
force_without_xla_support_flag = False,
tags = ["no_rocm"],
)
saved_model_compile_aot(
@ -466,7 +464,6 @@ saved_model_compile_aot(
"//tensorflow/cc/saved_model:saved_model_half_plus_two",
],
force_without_xla_support_flag = False,
tags = ["no_rocm"],
variables_to_feed = "variable_x",
)
@ -503,7 +500,6 @@ tf_cc_test(
srcs = if_xla_available([
"aot_compiled_test.cc",
]),
tags = ["no_rocm"],
deps = [
"//tensorflow/core:test_main",
] + if_xla_available([

1
tensorflow/python/tpu/BUILD
View File

@ -34,7 +34,6 @@ py_test(
"no_oss_py2",
"no_oss_py35",
"no_pip",
"no_rocm",
],
deps = [
"//tensorflow/python:client_testlib",

1
tensorflow/tools/compatibility/BUILD
View File

@ -270,7 +270,6 @@ py_test(
srcs = ["test_file_v2_0.py"],
python_version = "PY3",
srcs_version = "PY2AND3",
tags = ["no_rocm"],
deps = [
"//tensorflow:tensorflow_py",
],