Add python_version
arg to python test macros in tensorflow.bzl
.
PiperOrigin-RevId: 281518994 Change-Id: I14b8daf75cd75776a23864236f5b9f3262904385
This commit is contained in:
parent
2aa9a78d87
commit
b646d41f69
@ -2152,7 +2152,8 @@ def gpu_py_test(
|
|||||||
flaky = 0,
|
flaky = 0,
|
||||||
xla_enable_strict_auto_jit = False,
|
xla_enable_strict_auto_jit = False,
|
||||||
xla_enabled = 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
|
# TODO(b/122522101): Don't ignore xla_enable_strict_auto_jit and enable additional
|
||||||
# XLA tests once enough compute resources are available.
|
# XLA tests once enough compute resources are available.
|
||||||
_ignored = [xla_enable_strict_auto_jit]
|
_ignored = [xla_enable_strict_auto_jit]
|
||||||
@ -2179,6 +2180,7 @@ def gpu_py_test(
|
|||||||
tags = test_tags,
|
tags = test_tags,
|
||||||
xla_enabled = xla_enabled,
|
xla_enabled = xla_enabled,
|
||||||
xla_enable_strict_auto_jit = False,
|
xla_enable_strict_auto_jit = False,
|
||||||
|
python_version = python_version,
|
||||||
)
|
)
|
||||||
|
|
||||||
register_extension_info(
|
register_extension_info(
|
||||||
@ -2243,7 +2245,8 @@ def py_tests(
|
|||||||
prefix = "",
|
prefix = "",
|
||||||
xla_enable_strict_auto_jit = False,
|
xla_enable_strict_auto_jit = False,
|
||||||
xla_enabled = False,
|
xla_enabled = False,
|
||||||
grpc_enabled = False):
|
grpc_enabled = False,
|
||||||
|
python_version = "PY2"):
|
||||||
for src in srcs:
|
for src in srcs:
|
||||||
test_name = src.split("/")[-1].split(".")[0]
|
test_name = src.split("/")[-1].split(".")[0]
|
||||||
if prefix:
|
if prefix:
|
||||||
@ -2261,6 +2264,7 @@ def py_tests(
|
|||||||
tags = tags,
|
tags = tags,
|
||||||
xla_enabled = xla_enabled,
|
xla_enabled = xla_enabled,
|
||||||
xla_enable_strict_auto_jit = xla_enable_strict_auto_jit,
|
xla_enable_strict_auto_jit = xla_enable_strict_auto_jit,
|
||||||
|
python_version = python_version,
|
||||||
)
|
)
|
||||||
|
|
||||||
def gpu_py_tests(
|
def gpu_py_tests(
|
||||||
@ -2275,7 +2279,8 @@ def gpu_py_tests(
|
|||||||
prefix = "",
|
prefix = "",
|
||||||
xla_enable_strict_auto_jit = False,
|
xla_enable_strict_auto_jit = False,
|
||||||
xla_enabled = 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
|
# TODO(b/122522101): Don't ignore xla_enable_strict_auto_jit and enable additional
|
||||||
# XLA tests once enough compute resources are available.
|
# XLA tests once enough compute resources are available.
|
||||||
_ignored = [xla_enable_strict_auto_jit]
|
_ignored = [xla_enable_strict_auto_jit]
|
||||||
@ -2293,6 +2298,7 @@ def gpu_py_tests(
|
|||||||
tags = test_tags,
|
tags = test_tags,
|
||||||
xla_enabled = xla_enabled,
|
xla_enabled = xla_enabled,
|
||||||
xla_enable_strict_auto_jit = False,
|
xla_enable_strict_auto_jit = False,
|
||||||
|
python_version = python_version,
|
||||||
)
|
)
|
||||||
|
|
||||||
# terminology changes: saving cuda_* definition for compatibility
|
# terminology changes: saving cuda_* definition for compatibility
|
||||||
|
Loading…
Reference in New Issue
Block a user