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

Adjust collectives test size and change to tf_cuda_cc_test type.

Browse Source 
PiperOrigin-RevId: 326716294
Change-Id: I6454c05e7aba92c4195f675d1b8a51829ea70d63
This commit is contained in:
Ayush Dubey 2020-08-14 13:13:02 -07:00 committed by TensorFlower Gardener
parent 43cfb92ac5
commit 2f8072659a
1 changed files with 8 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
tensorflow/core/common_runtime/BUILD
View File

@ -13,9 +13,6 @@ load(
# buildifier: disable=same-origin-load
load("//tensorflow:tensorflow.bzl", "tf_cc_test_gpu")
# buildifier: disable=same-origin-load
load("//tensorflow:tensorflow.bzl", "tf_cc_tests_gpu")
# buildifier: disable=same-origin-load
load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test")
@ -1912,9 +1909,9 @@ tf_cc_tests(
],
)
tf_cc_tests_gpu(
tf_cuda_cc_test(
name = "ring_reducer_test",
size = "medium",
size = "small",
srcs = [
"ring_reducer_test.cc",
],
@ -1927,7 +1924,6 @@ tf_cc_tests_gpu(
"//tensorflow/core:all_kernels",
"//tensorflow/core:framework",
"//tensorflow/core:framework_internal",
"//tensorflow/core:gpu_runtime",
"//tensorflow/core:lib",
"//tensorflow/core:lib_internal",
"//tensorflow/core:ops",
@ -1935,14 +1931,13 @@ tf_cc_tests_gpu(
"//tensorflow/core:test",
"//tensorflow/core:test_main",
"//tensorflow/core:testlib",
"//tensorflow/core/util:protos_test_cc",
"@com_google_absl//absl/memory",
],
)
tf_cc_tests_gpu(
tf_cuda_cc_test(
name = "ring_gatherer_test",
size = "medium",
size = "small",
srcs = [
"ring_gatherer_test.cc",
],
@ -1962,15 +1957,13 @@ tf_cc_tests_gpu(
"//tensorflow/core:test",
"//tensorflow/core:test_main",
"//tensorflow/core:testlib",
"//tensorflow/core/common_runtime/gpu:gpu_runtime",
"//tensorflow/core/util:protos_test_cc",
"@com_google_absl//absl/memory",
],
)
tf_cc_tests_gpu(
tf_cuda_cc_test(
name = "hierarchical_tree_broadcaster_test",
size = "medium",
size = "small",
srcs = [
"hierarchical_tree_broadcaster_test.cc",
],
@ -1990,15 +1983,13 @@ tf_cc_tests_gpu(
"//tensorflow/core:test",
"//tensorflow/core:test_main",
"//tensorflow/core:testlib",
"//tensorflow/core/common_runtime/gpu:gpu_runtime",
"//tensorflow/core/util:protos_test_cc",
"@com_google_absl//absl/memory",
],
)
tf_cc_tests_gpu(
tf_cuda_cc_test(
name = "permuter_test",
size = "medium",
size = "small",
srcs = [
"permuter_test.cc",
],
@ -2018,8 +2009,6 @@ tf_cc_tests_gpu(
"//tensorflow/core:test",
"//tensorflow/core:test_main",
"//tensorflow/core:testlib",
"//tensorflow/core/common_runtime/gpu:gpu_runtime",
"//tensorflow/core/util:protos_test_cc",
"@com_google_absl//absl/memory",
],
)