removing no_rocm tag from tests that are now passing on the ROCm platform
This commit is contained in:
parent
e9e1e8e538
commit
c3a5543e17
@ -3651,9 +3651,6 @@ cuda_py_test(
|
||||
size = "small",
|
||||
srcs = ["training/experimental/mixed_precision_test.py"],
|
||||
python_version = "PY3",
|
||||
tags = [
|
||||
"no_rocm",
|
||||
],
|
||||
deps = [
|
||||
":mixed_precision",
|
||||
"//tensorflow/python:client_testlib",
|
||||
@ -4764,7 +4761,6 @@ cuda_py_test(
|
||||
srcs = ["ops/nn_fused_batchnorm_test.py"],
|
||||
python_version = "PY3",
|
||||
shard_count = 16,
|
||||
tags = ["no_rocm"],
|
||||
deps = [
|
||||
":array_ops",
|
||||
":client_testlib",
|
||||
|
@ -70,7 +70,6 @@ cuda_py_test(
|
||||
srcs = ["xla_test.py"],
|
||||
tags = [
|
||||
"no_mac",
|
||||
"no_rocm", # XLA support is not enabled on the ROCm platform
|
||||
"no_windows",
|
||||
],
|
||||
xla_enabled = True,
|
||||
|
@ -83,9 +83,6 @@ tf_py_test(
|
||||
name = "dataset_test",
|
||||
size = "small",
|
||||
srcs = ["dataset_test.py"],
|
||||
tags = [
|
||||
"no_rocm",
|
||||
],
|
||||
deps = [
|
||||
":test_base",
|
||||
"//tensorflow/core:protos_all_py",
|
||||
|
@ -715,7 +715,6 @@ cuda_py_test(
|
||||
tags = [
|
||||
"guitar",
|
||||
"multi_and_single_gpu",
|
||||
"no_rocm",
|
||||
"no_windows", # TODO(b/142475891): Enable this test on Windows.
|
||||
"no_windows_gpu", # TODO(b/130551176)
|
||||
],
|
||||
|
@ -172,9 +172,6 @@ py_test(
|
||||
srcs = ["distribute_lib_test.py"],
|
||||
python_version = "PY3",
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_rocm",
|
||||
],
|
||||
deps = [
|
||||
":combinations",
|
||||
":distribute_lib",
|
||||
@ -941,7 +938,6 @@ distribute_py_test(
|
||||
tags = [
|
||||
"multi_and_single_gpu",
|
||||
"no_oss", # TODO(b/139815303): enable after this is fixed.
|
||||
"no_rocm",
|
||||
"notap", # TODO(b/139815303): enable after this is fixed.
|
||||
],
|
||||
deps = [
|
||||
@ -995,7 +991,6 @@ distribute_py_test(
|
||||
main = "step_fn_test.py",
|
||||
tags = [
|
||||
"multi_and_single_gpu",
|
||||
"no_rocm",
|
||||
],
|
||||
deps = [
|
||||
":single_loss_example",
|
||||
|
@ -266,7 +266,6 @@ cuda_py_test(
|
||||
srcs = ["backprop_test.py"],
|
||||
python_version = "PY3",
|
||||
tags = [
|
||||
"no_rocm",
|
||||
"no_windows", #TODO(b/139745667)
|
||||
],
|
||||
deps = [
|
||||
|
@ -772,7 +772,6 @@ tf_py_test(
|
||||
srcs = ["layers/convolutional_recurrent_test.py"],
|
||||
python_version = "PY3",
|
||||
shard_count = 4,
|
||||
tags = ["no_rocm"],
|
||||
deps = [
|
||||
":keras",
|
||||
"//tensorflow/python:client_testlib",
|
||||
@ -829,7 +828,6 @@ cuda_py_test(
|
||||
python_version = "PY3",
|
||||
shard_count = 4,
|
||||
tags = [
|
||||
"no_rocm",
|
||||
"no_windows_gpu",
|
||||
],
|
||||
deps = [
|
||||
@ -972,7 +970,6 @@ cuda_py_test(
|
||||
python_version = "PY3",
|
||||
shard_count = 4,
|
||||
tags = [
|
||||
"no_rocm",
|
||||
"notsan",
|
||||
],
|
||||
deps = [
|
||||
@ -1580,7 +1577,6 @@ tf_py_test(
|
||||
srcs = ["engine/training_arrays_test.py"],
|
||||
python_version = "PY3",
|
||||
tags = [
|
||||
"no_rocm",
|
||||
"nomac", # TODO(mihaimaruseac): b/127695564
|
||||
],
|
||||
deps = [
|
||||
@ -1655,7 +1651,6 @@ tf_py_test(
|
||||
srcs = ["engine/training_eager_test.py"],
|
||||
python_version = "PY3",
|
||||
tags = [
|
||||
"no_rocm",
|
||||
"nomac", # TODO(mihaimaruseac): b/127695564
|
||||
"notsan",
|
||||
],
|
||||
@ -1805,7 +1800,6 @@ tf_py_test(
|
||||
srcs = ["engine/base_layer_utils_test.py"],
|
||||
python_version = "PY3",
|
||||
tags = [
|
||||
"no_rocm",
|
||||
"nomac", # TODO(mihaimaruseac): b/127695564
|
||||
],
|
||||
deps = [
|
||||
|
@ -199,7 +199,6 @@ distribute_py_test(
|
||||
shard_count = 4,
|
||||
tags = [
|
||||
"multi_and_single_gpu",
|
||||
"no_rocm", # times out on ROCm
|
||||
"no_windows_gpu",
|
||||
"notsan",
|
||||
],
|
||||
|
@ -102,7 +102,6 @@ cuda_py_test(
|
||||
size = "medium",
|
||||
srcs = ["adamax_test.py"],
|
||||
shard_count = 4,
|
||||
tags = ["no_rocm"],
|
||||
deps = [
|
||||
":optimizer_v2",
|
||||
"//tensorflow/python:client_testlib",
|
||||
@ -201,7 +200,6 @@ cuda_py_test(
|
||||
tags = [
|
||||
"manual",
|
||||
"no_oss",
|
||||
"no_rocm",
|
||||
"no_windows",
|
||||
"notap",
|
||||
],
|
||||
|
@ -120,7 +120,6 @@ cuda_py_test(
|
||||
size = "small",
|
||||
srcs = ["list_ops_test.py"],
|
||||
grpc_enabled = True,
|
||||
tags = ["no_rocm"],
|
||||
deps = [
|
||||
"//tensorflow/python:array_ops",
|
||||
"//tensorflow/python:client_testlib",
|
||||
@ -1287,7 +1286,6 @@ cuda_py_test(
|
||||
shard_count = 10,
|
||||
tags = [
|
||||
"no_oss", # TODO(b/142818120): Re-enable.
|
||||
"no_rocm",
|
||||
],
|
||||
deps = [
|
||||
"//tensorflow/python:client_testlib",
|
||||
@ -2717,7 +2715,6 @@ cuda_py_test(
|
||||
srcs = ["tensor_array_ops_test.py"],
|
||||
flaky = 1, # create_local_cluster sometimes times out.
|
||||
shard_count = 10,
|
||||
tags = ["no_rocm"],
|
||||
deps = [
|
||||
"//tensorflow/python:array_ops",
|
||||
"//tensorflow/python:client_testlib",
|
||||
@ -3448,9 +3445,6 @@ cuda_py_test(
|
||||
size = "medium",
|
||||
srcs = ["qr_op_test.py"],
|
||||
shard_count = 20,
|
||||
tags = [
|
||||
"no_rocm", # TODO(rocm): feature not supported on ROCm platform
|
||||
],
|
||||
deps = [
|
||||
"//tensorflow/python:array_ops",
|
||||
"//tensorflow/python:client_testlib",
|
||||
|
@ -25,7 +25,6 @@ cuda_py_tests(
|
||||
name = "dct_ops_test",
|
||||
srcs = ["dct_ops_test.py"],
|
||||
python_version = "PY3",
|
||||
tags = ["no_rocm"],
|
||||
deps = [
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:framework_for_generated_wrappers",
|
||||
@ -70,7 +69,6 @@ cuda_py_tests(
|
||||
name = "mfcc_ops_test",
|
||||
srcs = ["mfcc_ops_test.py"],
|
||||
python_version = "PY3",
|
||||
tags = ["no_rocm"],
|
||||
deps = [
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:framework",
|
||||
|
@ -82,7 +82,6 @@ py_test(
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_pip",
|
||||
"no_rocm",
|
||||
],
|
||||
deps = [
|
||||
"//tensorflow/python:client_testlib",
|
||||
|
Loading…
Reference in New Issue
Block a user