diff options
author | Derek Murray <mrry@google.com> | 2017-12-11 07:58:48 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-11 08:02:21 -0800 |
commit | 6c33765dd952ff0f9d1de9e3f81f972a9489f319 (patch) | |
tree | 3bb04d6857720610720d19ec1c5362744ba194b0 /tensorflow/tensorflow.bzl | |
parent | 218caf995a1f1ea4883ed5575f7cd70536eff60c (diff) |
Add `grpc_enabled` optional argument to various Python test rules.
PiperOrigin-RevId: 178618409
Diffstat (limited to 'tensorflow/tensorflow.bzl')
-rw-r--r-- | tensorflow/tensorflow.bzl | 30 |
1 files changed, 21 insertions, 9 deletions
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl index 0db915f1b9..425997e41f 100644 --- a/tensorflow/tensorflow.bzl +++ b/tensorflow/tensorflow.bzl @@ -6,6 +6,7 @@ load( "//tensorflow/core:platform/default/build_config_root.bzl", "tf_cuda_tests_tags", "tf_sycl_tests_tags", + "tf_additional_grpc_deps_py", "tf_additional_xla_deps_py", "if_static", ) @@ -1340,9 +1341,12 @@ def tf_py_test(name, shard_count=1, additional_deps=[], flaky=0, - xla_enabled=False): + xla_enabled=False, + grpc_enabled=False): if xla_enabled: additional_deps = additional_deps + tf_additional_xla_deps_py() + if grpc_enabled: + additional_deps = additional_deps + tf_additional_grpc_deps_py() native.py_test( name=name, size=size, @@ -1378,7 +1382,8 @@ def cuda_py_test(name, additional_deps=[], tags=[], flaky=0, - xla_enabled=False): + xla_enabled=False, + grpc_enabled=False): test_tags = tags + tf_cuda_tests_tags() tf_py_test( name=name, @@ -1391,7 +1396,8 @@ def cuda_py_test(name, shard_count=shard_count, additional_deps=additional_deps, flaky=flaky, - xla_enabled=xla_enabled) + xla_enabled=xla_enabled, + grpc_enabled=grpc_enabled) register_extension_info( extension_name = "cuda_py_test", @@ -1408,7 +1414,8 @@ def sycl_py_test(name, additional_deps=[], tags=[], flaky=0, - xla_enabled=False): + xla_enabled=False, + grpc_enabled=False): test_tags = tags + tf_sycl_tests_tags() tf_py_test( name=name, @@ -1421,7 +1428,8 @@ def sycl_py_test(name, shard_count=shard_count, additional_deps=additional_deps, flaky=flaky, - xla_enabled=xla_enabled) + xla_enabled=xla_enabled, + grpc_enabled=grpc_enabled) register_extension_info( extension_name = "sycl_py_test", @@ -1436,7 +1444,8 @@ def py_tests(name, tags=[], shard_count=1, prefix="", - xla_enabled=False): + xla_enabled=False, + grpc_enabled=False): for src in srcs: test_name = src.split("/")[-1].split(".")[0] if prefix: @@ -1450,7 +1459,8 @@ def py_tests(name, shard_count=shard_count, data=data, additional_deps=additional_deps, - xla_enabled=xla_enabled) + xla_enabled=xla_enabled, + grpc_enabled=grpc_enabled) def cuda_py_tests(name, srcs, @@ -1460,7 +1470,8 @@ def cuda_py_tests(name, shard_count=1, tags=[], prefix="", - xla_enabled=False): + xla_enabled=False, + grpc_enabled=False): test_tags = tags + tf_cuda_tests_tags() py_tests( name=name, @@ -1471,7 +1482,8 @@ def cuda_py_tests(name, tags=test_tags, shard_count=shard_count, prefix=prefix, - xla_enabled=xla_enabled) + xla_enabled=xla_enabled, + grpc_enabled=grpc_enabled) # Creates a genrule named <name> for running tools/proto_text's generator to # make the proto_text functions, for the protos passed in <srcs>. |