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

Add python_version arg to python test macros in tensorflow.bzl.

Browse Source 
PiperOrigin-RevId: 281518994
Change-Id: I14b8daf75cd75776a23864236f5b9f3262904385
This commit is contained in:
Hye Soo Yang 2019-11-20 07:49:33 -08:00 committed by TensorFlower Gardener
parent 2aa9a78d87
commit b646d41f69
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tensorflow/tensorflow.bzl
View File

@ -2152,7 +2152,8 @@ def gpu_py_test(
flaky = 0,
xla_enable_strict_auto_jit = False,
xla_enabled = False,
grpc_enabled = False):
grpc_enabled = False,
python_version = "PY2"):
# TODO(b/122522101): Don't ignore xla_enable_strict_auto_jit and enable additional
# XLA tests once enough compute resources are available.
_ignored = [xla_enable_strict_auto_jit]
@ -2179,6 +2180,7 @@ def gpu_py_test(
tags = test_tags,
xla_enabled = xla_enabled,
xla_enable_strict_auto_jit = False,
python_version = python_version,
)
register_extension_info(
@ -2243,7 +2245,8 @@ def py_tests(
prefix = "",
xla_enable_strict_auto_jit = False,
xla_enabled = False,
grpc_enabled = False):
grpc_enabled = False,
python_version = "PY2"):
for src in srcs:
test_name = src.split("/")[-1].split(".")[0]
if prefix:
@ -2261,6 +2264,7 @@ def py_tests(
tags = tags,
xla_enabled = xla_enabled,
xla_enable_strict_auto_jit = xla_enable_strict_auto_jit,
python_version = python_version,
)
def gpu_py_tests(
@ -2275,7 +2279,8 @@ def gpu_py_tests(
prefix = "",
xla_enable_strict_auto_jit = False,
xla_enabled = False,
grpc_enabled = False):
grpc_enabled = False,
python_version = "PY2"):
# TODO(b/122522101): Don't ignore xla_enable_strict_auto_jit and enable additional
# XLA tests once enough compute resources are available.
_ignored = [xla_enable_strict_auto_jit]
@ -2293,6 +2298,7 @@ def gpu_py_tests(
tags = test_tags,
xla_enabled = xla_enabled,
xla_enable_strict_auto_jit = False,
python_version = python_version,
)
# terminology changes: saving cuda_* definition for compatibility