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

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

Browse Source
This commit is contained in:
Deven Desai 2021-01-19 14:48:33 +00:00
parent de892f9bc7
commit 4871a1c963
9 changed files with 4 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/compiler/tests/BUILD
View File

@ -1048,7 +1048,6 @@ tf_xla_py_test(
shard_count = 5,
tags = [
"no_pip", # TODO(b/149738646): fix pip install so these tests run on kokoro pip
"no_rocm",
"optonly",
],
deps = [

1
tensorflow/compiler/xla/client/lib/BUILD
View File

@ -609,7 +609,6 @@ xla_test(
name = "logdet_test",
srcs = ["logdet_test.cc"],
tags = [
"no_rocm",
"optonly",
],
deps = [

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

@ -177,7 +177,7 @@ tf_cc_test(
srcs = [
"tree_reduction_rewriter_test.cc",
],
tags = tf_cuda_tests_tags() + ["no_rocm"],
tags = tf_cuda_tests_tags(),
deps = [
":gpu_codegen_test",
"//tensorflow/compiler/xla:debug_options_flags",
@ -258,7 +258,7 @@ tf_cc_test(
srcs = [
"parallel_reduction_test.cc",
],
tags = tf_cuda_tests_tags() + ["no_rocm"],
tags = tf_cuda_tests_tags(),
deps = [
":gpu_codegen_test",
"//tensorflow/compiler/xla/service:gpu_plugin",
@ -297,7 +297,7 @@ tf_cc_test(
srcs = [
"gpu_copy_alone_test.cc",
],
tags = tf_cuda_tests_tags() + ["no_rocm"],
tags = tf_cuda_tests_tags(),
deps = [
":gpu_codegen_test",
"//tensorflow/compiler/xla/service:hlo",
@ -521,9 +521,7 @@ tf_cc_test(
srcs = [
"sorting_test.cc",
],
tags = tf_cuda_tests_tags() + [
"no_rocm",
],
tags = tf_cuda_tests_tags(),
deps = [
":gpu_codegen_test",
"//tensorflow/compiler/xla:debug_options_flags",

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

@ -1159,7 +1159,6 @@ xla_test(
],
shard_count = 50,
tags = [
"no_rocm",
"optonly",
],
deps = CONVOLUTION_TEST_DEPS + [
@ -1212,9 +1211,6 @@ xla_test(
backend_args = {"gpu": ["--xla_backend_extra_options=xla_gpu_experimental_conv_disable_layout_heuristic"]},
backends = ["gpu"],
shard_count = 25,
tags = [
"no_rocm",
],
deps = CONVOLUTION_TEST_DEPS + [
"@com_google_absl//absl/memory",
"@com_google_absl//absl/strings",
@ -1228,9 +1224,6 @@ xla_test(
backend_args = {"gpu": ["--xla_backend_extra_options=xla_gpu_experimental_conv_disable_layout_heuristic"]},
backends = ["gpu"],
shard_count = 25,
tags = [
"no_rocm",
],
deps = CONVOLUTION_TEST_DEPS + [
"@com_google_absl//absl/memory",
"@com_google_absl//absl/strings",

1
tensorflow/python/BUILD
View File

@ -3103,7 +3103,6 @@ cuda_py_test(
tags = [
"guitar",
"multi_gpu",
"no_rocm",
"no_windows",
],
deps = [

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

@ -80,7 +80,6 @@ cuda_py_test(
name = "gradient_checkpoint_test",
srcs = ["gradient_checkpoint_test.py"],
python_version = "PY3",
tags = ["no_rocm"],
deps = [
"//tensorflow:tensorflow_py_no_contrib",
],

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

@ -158,7 +158,6 @@ cuda_py_test(
size = "medium",
srcs = ["adadelta_test.py"],
shard_count = 4,
tags = ["no_rocm"],
# TODO(b/168527439): invalid resource variable reference on GPU for TFRT.
deps = [
":optimizer_v2",
@ -239,7 +238,6 @@ cuda_py_test(
srcs = ["optimizer_v2_test.py"],
shard_count = 8,
tags = [
"no_rocm",
"no_windows",
],
deps = [
@ -297,7 +295,6 @@ cuda_py_test(
size = "medium",
srcs = ["rmsprop_test.py"],
shard_count = 2,
tags = ["no_rocm"],
xla_tags = [
"no_cuda_asan", # times out
],

1
tensorflow/python/profiler/integration_test/BUILD
View File

@ -21,7 +21,6 @@ cuda_py_test(
python_version = "PY3",
tags = [
"no_pip",
"no_rocm",
],
deps = [
":mnist_testing_utils",

1
tensorflow/tools/docs/BUILD
View File

@ -46,7 +46,6 @@ py_test(
tags = [
"no_oss_py2",
"no_pip",
"no_rocm",
"no_windows", # numpy prints differently on windows.
"noasan",
"nomsan",