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

Removing no_rocm tag from unit-tests that are now passing on the ROCm platform

Browse Source
This commit is contained in:
Deven Desai 2020-11-24 21:15:18 +00:00
parent 6a30edd2de
commit 6ccddc5d3f
21 changed files with 1 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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",
@ -1627,7 +1626,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,
@ -1652,7 +1650,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,
@ -1747,7 +1744,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,
@ -1871,7 +1867,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",

2
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",

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

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

@ -626,7 +626,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 +664,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",
@ -1514,7 +1512,6 @@ xla_test(
srcs = ["reduce_test.cc"],
shard_count = 31,
tags = [
"no_rocm",
"optonly",
],
deps = [
@ -1594,7 +1591,6 @@ xla_test(
timeout = "long",
srcs = ["select_and_scatter_test.cc"],
tags = [
"no_rocm",
"nozapfhahn",
"optonly",
],

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 = [

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

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

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

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

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

@ -389,7 +389,6 @@ tf_py_test(
shard_count = 20,
tags = [
"manual",
"no_rocm",
"nomac", # TODO(mihaimaruseac): b/127695564
"notsan",
],
@ -514,7 +513,6 @@ tf_py_test(
python_version = "PY3",
shard_count = 30,
tags = [
"no_rocm",
"nomac", # TODO(mihaimaruseac): b/127695564
],
deps = [
@ -658,7 +656,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 = [

2
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 = [
@ -3847,7 +3846,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/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
],

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",
],