Move additional_deps to deps for cuda_py_test.
PiperOrigin-RevId: 285289102 Change-Id: Ic1ca096f27689fda42c9fc98505adbe5b7489993
This commit is contained in:
Brian Atkinson
TensorFlower Gardener
parent
e63ae8db22
commit
13088cc7ba
@ -108,11 +108,11 @@ cuda_py_test(
|
||||
name = "cancellation_test",
|
||||
size = "small",
|
||||
srcs = ["cancellation_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
deps = [
|
||||
":cancellation",
|
||||
":test",
|
||||
],
|
||||
python_version = "PY3",
|
||||
)
|
||||
|
||||
py_library(
|
||||
@ -148,11 +148,11 @@ cuda_py_test(
|
||||
name = "context_test",
|
||||
size = "small",
|
||||
srcs = ["context_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
deps = [
|
||||
":context",
|
||||
":test",
|
||||
],
|
||||
python_version = "PY3",
|
||||
)
|
||||
|
||||
py_library(
|
||||
@ -173,11 +173,11 @@ py_library(
|
||||
cuda_py_test(
|
||||
name = "monitoring_test",
|
||||
srcs = ["monitoring_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
deps = [
|
||||
":monitoring",
|
||||
":test",
|
||||
],
|
||||
python_version = "PY3",
|
||||
)
|
||||
|
||||
py_library(
|
||||
@ -195,12 +195,12 @@ py_library(
|
||||
cuda_py_test(
|
||||
name = "profiler_test",
|
||||
srcs = ["profiler_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
deps = [
|
||||
":profiler",
|
||||
":test",
|
||||
"//tensorflow/python:constant_op",
|
||||
],
|
||||
python_version = "PY3",
|
||||
)
|
||||
|
||||
py_library(
|
||||
@ -236,141 +236,142 @@ py_library(
|
||||
cuda_py_test(
|
||||
name = "tensor_test",
|
||||
srcs = ["tensor_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
deps = [
|
||||
":context",
|
||||
":test",
|
||||
"//tensorflow/python:errors",
|
||||
"//tensorflow/python:framework_ops",
|
||||
],
|
||||
python_version = "PY3",
|
||||
)
|
||||
|
||||
cuda_py_test(
|
||||
name = "backprop_test",
|
||||
srcs = ["backprop_test.py"],
|
||||
additional_deps = [
|
||||
":backprop",
|
||||
":context",
|
||||
":test",
|
||||
"//third_party/py/numpy",
|
||||
"//tensorflow/python:embedding_ops",
|
||||
"//tensorflow/python:array_ops",
|
||||
"//tensorflow/python:control_flow_ops",
|
||||
"//tensorflow/python:layers",
|
||||
"//tensorflow/python:math_ops",
|
||||
"//tensorflow/python:nn_ops",
|
||||
"//tensorflow/python:resource_variable_ops",
|
||||
"//tensorflow/python:random_ops",
|
||||
"//tensorflow/python:nn_grad",
|
||||
"//tensorflow/python:training",
|
||||
],
|
||||
python_version = "PY3",
|
||||
tags = [
|
||||
"no_rocm",
|
||||
"no_windows", #TODO(b/139745667)
|
||||
],
|
||||
deps = [
|
||||
":backprop",
|
||||
":context",
|
||||
":test",
|
||||
"//tensorflow/python:array_ops",
|
||||
"//tensorflow/python:control_flow_ops",
|
||||
"//tensorflow/python:embedding_ops",
|
||||
"//tensorflow/python:layers",
|
||||
"//tensorflow/python:math_ops",
|
||||
"//tensorflow/python:nn_grad",
|
||||
"//tensorflow/python:nn_ops",
|
||||
"//tensorflow/python:random_ops",
|
||||
"//tensorflow/python:resource_variable_ops",
|
||||
"//tensorflow/python:training",
|
||||
"//third_party/py/numpy",
|
||||
],
|
||||
)
|
||||
|
||||
cuda_py_test(
|
||||
name = "forwardprop_test",
|
||||
srcs = ["forwardprop_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
shard_count = 5,
|
||||
deps = [
|
||||
":forwardprop",
|
||||
":forwardprop_util",
|
||||
":test",
|
||||
"//tensorflow/python/distribute:mirrored_strategy",
|
||||
],
|
||||
python_version = "PY3",
|
||||
shard_count = 5,
|
||||
)
|
||||
|
||||
cuda_py_test(
|
||||
name = "core_test",
|
||||
size = "small",
|
||||
srcs = ["core_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
deps = [
|
||||
":context",
|
||||
":core",
|
||||
":execute",
|
||||
":test",
|
||||
"//third_party/py/numpy",
|
||||
"//tensorflow/python:dtypes",
|
||||
"//tensorflow/python:errors",
|
||||
"//tensorflow/python:framework_ops",
|
||||
"//tensorflow/python:framework_test_lib",
|
||||
"//tensorflow/python:pywrap_tensorflow",
|
||||
"//third_party/py/numpy",
|
||||
],
|
||||
python_version = "PY3",
|
||||
)
|
||||
|
||||
cuda_py_test(
|
||||
name = "function_argument_naming_test",
|
||||
size = "medium",
|
||||
srcs = ["function_argument_naming_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
deps = [
|
||||
":backprop",
|
||||
":def_function",
|
||||
":function",
|
||||
":test",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
"//tensorflow/python:math_ops",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
],
|
||||
python_version = "PY3",
|
||||
)
|
||||
|
||||
cuda_py_test(
|
||||
name = "function_defun_collection_test",
|
||||
size = "medium",
|
||||
srcs = ["function_defun_collection_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
deps = [
|
||||
":backprop",
|
||||
":def_function",
|
||||
":function",
|
||||
":test",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
"//tensorflow/python:math_ops",
|
||||
"//tensorflow/python:resource_variable_ops",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
],
|
||||
python_version = "PY3",
|
||||
)
|
||||
|
||||
cuda_py_test(
|
||||
name = "function_gradients_test",
|
||||
size = "medium",
|
||||
srcs = ["function_gradients_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
shard_count = 5,
|
||||
deps = [
|
||||
":backprop",
|
||||
":context",
|
||||
":def_function",
|
||||
":function",
|
||||
":test",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
"//tensorflow/python:math_ops",
|
||||
"//tensorflow/python:resource_variable_ops",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
],
|
||||
python_version = "PY3",
|
||||
shard_count = 5,
|
||||
)
|
||||
|
||||
cuda_py_test(
|
||||
name = "function_test",
|
||||
size = "medium",
|
||||
srcs = ["function_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
shard_count = 15,
|
||||
deps = [
|
||||
":backprop",
|
||||
":cancellation",
|
||||
":context",
|
||||
":def_function",
|
||||
":function",
|
||||
":test",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
"//tensorflow/python:clip_ops",
|
||||
"//tensorflow/python:constant_op",
|
||||
"//tensorflow/python:data_flow_ops",
|
||||
"//tensorflow/python:dtypes",
|
||||
"//tensorflow/python:gradients",
|
||||
"//tensorflow/python:errors",
|
||||
"//tensorflow/python:framework_ops",
|
||||
"//tensorflow/python:gradients",
|
||||
"//tensorflow/python:indexed_slices",
|
||||
"//tensorflow/python:init_ops",
|
||||
"//tensorflow/python:layers",
|
||||
@ -382,9 +383,8 @@ cuda_py_test(
|
||||
"//tensorflow/python:tensor_shape",
|
||||
"//tensorflow/python:tensor_spec",
|
||||
"//tensorflow/python:test_ops",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
],
|
||||
python_version = "PY3",
|
||||
shard_count = 15,
|
||||
)
|
||||
|
||||
py_library(
|
||||
@ -432,15 +432,15 @@ py_library(
|
||||
cuda_py_test(
|
||||
name = "graph_only_ops_test",
|
||||
srcs = ["graph_only_ops_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
deps = [
|
||||
"graph_only_ops",
|
||||
"//third_party/py/numpy",
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:dtypes",
|
||||
"//tensorflow/python:framework_test_lib",
|
||||
"//tensorflow/python:math_ops",
|
||||
"//third_party/py/numpy",
|
||||
],
|
||||
python_version = "PY3",
|
||||
)
|
||||
|
||||
py_library(
|
||||
@ -542,39 +542,39 @@ py_library(
|
||||
cuda_py_test(
|
||||
name = "benchmarks_test",
|
||||
srcs = ["benchmarks_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
deps = [
|
||||
":backprop",
|
||||
":context",
|
||||
":forwardprop",
|
||||
":function",
|
||||
":test",
|
||||
":profiler",
|
||||
":remote",
|
||||
"//third_party/py/numpy",
|
||||
":test",
|
||||
"//tensorflow/python:math_ops",
|
||||
"//tensorflow/python:pywrap_tensorflow",
|
||||
"//tensorflow/python:random_ops",
|
||||
"//tensorflow/python/keras",
|
||||
"//third_party/py/numpy",
|
||||
],
|
||||
python_version = "PY3",
|
||||
)
|
||||
|
||||
cuda_py_test(
|
||||
name = "remote_benchmarks_test",
|
||||
srcs = ["remote_benchmarks_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
deps = [
|
||||
":backprop",
|
||||
":context",
|
||||
":forwardprop",
|
||||
":function",
|
||||
":test",
|
||||
":profiler",
|
||||
":remote",
|
||||
":test",
|
||||
"//tensorflow/python:math_ops",
|
||||
"//tensorflow/python:pywrap_tensorflow",
|
||||
"//tensorflow/python:random_ops",
|
||||
],
|
||||
python_version = "PY3",
|
||||
)
|
||||
|
||||
tf_py_logged_benchmark(
|
||||
@ -604,11 +604,11 @@ tf_py_test(
|
||||
cuda_py_test(
|
||||
name = "ops_test",
|
||||
srcs = ["ops_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
deps = [
|
||||
":context",
|
||||
":execute",
|
||||
":test",
|
||||
"//third_party/py/numpy",
|
||||
"//tensorflow/python:array_ops",
|
||||
"//tensorflow/python:config",
|
||||
"//tensorflow/python:control_flow_ops",
|
||||
@ -622,8 +622,8 @@ cuda_py_test(
|
||||
"//tensorflow/python:resource_variable_ops",
|
||||
"//tensorflow/python:sparse_ops",
|
||||
"//tensorflow/python:tensor_shape",
|
||||
"//third_party/py/numpy",
|
||||
],
|
||||
python_version = "PY3",
|
||||
)
|
||||
|
||||
tf_py_test(
|
||||
@ -704,21 +704,27 @@ tf_py_test(
|
||||
cuda_py_test(
|
||||
name = "def_function_test",
|
||||
srcs = ["def_function_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
deps = [
|
||||
":def_function",
|
||||
"//tensorflow/python/autograph/core",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:constant_op",
|
||||
"//tensorflow/python:framework_ops",
|
||||
"//tensorflow/python/autograph/core",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
],
|
||||
python_version = "PY3",
|
||||
)
|
||||
|
||||
cuda_py_test(
|
||||
name = "def_function_xla_jit_test",
|
||||
srcs = ["def_function_xla_jit_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
tags = [
|
||||
"no_mac",
|
||||
"no_windows",
|
||||
],
|
||||
xla_enabled = True,
|
||||
deps = [
|
||||
":backprop",
|
||||
":def_function",
|
||||
"//tensorflow/python:client_testlib",
|
||||
@ -726,12 +732,6 @@ cuda_py_test(
|
||||
"//tensorflow/python:framework_ops",
|
||||
"//tensorflow/python:resource_variable_ops",
|
||||
],
|
||||
python_version = "PY3",
|
||||
tags = [
|
||||
"no_mac",
|
||||
"no_windows",
|
||||
],
|
||||
xla_enabled = True,
|
||||
)
|
||||
|
||||
tf_xla_py_test(
|
||||
@ -798,15 +798,6 @@ cuda_py_test(
|
||||
name = "remote_test",
|
||||
size = "medium",
|
||||
srcs = ["remote_test.py"],
|
||||
additional_deps = [
|
||||
":context",
|
||||
":def_function",
|
||||
":test",
|
||||
":remote",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
"@six_archive//:six",
|
||||
"//tensorflow/python:resource_variable_ops",
|
||||
],
|
||||
grpc_enabled = True,
|
||||
python_version = "PY3",
|
||||
shard_count = 2,
|
||||
@ -814,6 +805,15 @@ cuda_py_test(
|
||||
"no_oss", # This test launches local server.
|
||||
"optonly", # times out
|
||||
],
|
||||
deps = [
|
||||
":context",
|
||||
":def_function",
|
||||
":remote",
|
||||
":test",
|
||||
"//tensorflow/python:resource_variable_ops",
|
||||
"@absl_py//absl/testing:parameterized",
|
||||
"@six_archive//:six",
|
||||
],
|
||||
)
|
||||
|
||||
tpu_py_test(
|
||||
@ -835,12 +835,12 @@ cuda_py_test(
|
||||
name = "device_placement_test",
|
||||
size = "small",
|
||||
srcs = ["device_placement_test.py"],
|
||||
additional_deps = [
|
||||
python_version = "PY3",
|
||||
deps = [
|
||||
":context",
|
||||
":def_function",
|
||||
":test",
|
||||
":remote",
|
||||
":test",
|
||||
"//tensorflow/python:resource_variable_ops",
|
||||
],
|
||||
python_version = "PY3",
|
||||
)
|
||||
|
||||
@ -39,11 +39,6 @@ cuda_py_test(
|
||||
name = "resnet50_test",
|
||||
size = "medium",
|
||||
srcs = ["resnet50_test.py"],
|
||||
additional_deps = [
|
||||
":resnet50",
|
||||
":resnet50_test_util",
|
||||
"//tensorflow:tensorflow_py_no_contrib",
|
||||
],
|
||||
shard_count = 4,
|
||||
tags = [
|
||||
"no_windows", # TODO(b/141617449): needs investigation
|
||||
@ -51,18 +46,17 @@ cuda_py_test(
|
||||
"oss_serial",
|
||||
"v1only",
|
||||
],
|
||||
deps = [
|
||||
":resnet50",
|
||||
":resnet50_test_util",
|
||||
"//tensorflow:tensorflow_py_no_contrib",
|
||||
],
|
||||
)
|
||||
|
||||
cuda_py_test(
|
||||
name = "hvp_test",
|
||||
size = "medium",
|
||||
srcs = ["hvp_test.py"],
|
||||
additional_deps = [
|
||||
":resnet50",
|
||||
":resnet50_test_util",
|
||||
"//tensorflow/python/eager:forwardprop",
|
||||
"//tensorflow:tensorflow_py_no_contrib",
|
||||
],
|
||||
shard_count = 7,
|
||||
tags = [
|
||||
"no_windows", # TODO(b/141617449): needs investigation
|
||||
@ -72,18 +66,18 @@ cuda_py_test(
|
||||
],
|
||||
# Times out
|
||||
xla_enable_strict_auto_jit = False,
|
||||
deps = [
|
||||
":resnet50",
|
||||
":resnet50_test_util",
|
||||
"//tensorflow:tensorflow_py_no_contrib",
|
||||
"//tensorflow/python/eager:forwardprop",
|
||||
],
|
||||
)
|
||||
|
||||
cuda_py_test(
|
||||
name = "resnet50_graph_test",
|
||||
size = "medium",
|
||||
srcs = ["resnet50_graph_test.py"],
|
||||
additional_deps = [
|
||||
":resnet50",
|
||||
":resnet50_test_lib",
|
||||
"//third_party/py/numpy",
|
||||
"//tensorflow:tensorflow_py_no_contrib",
|
||||
],
|
||||
shard_count = 4,
|
||||
tags = [
|
||||
"no_windows", # TODO(b/141617449): needs investigation
|
||||
@ -93,4 +87,10 @@ cuda_py_test(
|
||||
"optonly",
|
||||
"oss_serial",
|
||||
],
|
||||
deps = [
|
||||
":resnet50",
|
||||
":resnet50_test_lib",
|
||||
"//tensorflow:tensorflow_py_no_contrib",
|
||||
"//third_party/py/numpy",
|
||||
],
|
||||
)
|
||||
|
||||
@ -19,16 +19,6 @@ cuda_py_test(
|
||||
name = "memory_test",
|
||||
size = "medium",
|
||||
srcs = ["memory_test.py"],
|
||||
additional_deps = [
|
||||
":memory_test_util",
|
||||
"//tensorflow/python/keras",
|
||||
"//tensorflow/python/eager:backprop",
|
||||
"//tensorflow/python/eager:test",
|
||||
"//tensorflow/python:array_ops",
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:framework_test_lib",
|
||||
"@six_archive//:six",
|
||||
],
|
||||
tags = [
|
||||
"manual",
|
||||
"no_oss",
|
||||
@ -37,22 +27,32 @@ cuda_py_test(
|
||||
],
|
||||
# TODO(b/140065350): Re-enable
|
||||
xla_enable_strict_auto_jit = False,
|
||||
deps = [
|
||||
":memory_test_util",
|
||||
"//tensorflow/python:array_ops",
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python:framework_test_lib",
|
||||
"//tensorflow/python/eager:backprop",
|
||||
"//tensorflow/python/eager:test",
|
||||
"//tensorflow/python/keras",
|
||||
"@six_archive//:six",
|
||||
],
|
||||
)
|
||||
|
||||
cuda_py_test(
|
||||
name = "remote_memory_test",
|
||||
size = "medium",
|
||||
srcs = ["remote_memory_test.py"],
|
||||
additional_deps = [
|
||||
":memory_test_util",
|
||||
"//tensorflow/python/eager:backprop",
|
||||
"//tensorflow/python/eager:remote",
|
||||
"//tensorflow/python/eager:test",
|
||||
"//tensorflow/python:array_ops",
|
||||
"//tensorflow/python:client_testlib",
|
||||
],
|
||||
tags = [
|
||||
"optonly", # The test is too slow in non-opt mode
|
||||
],
|
||||
xla_enable_strict_auto_jit = False, # b/140261762
|
||||
deps = [
|
||||
":memory_test_util",
|
||||
"//tensorflow/python:array_ops",
|
||||
"//tensorflow/python:client_testlib",
|
||||
"//tensorflow/python/eager:backprop",
|
||||
"//tensorflow/python/eager:remote",
|
||||
"//tensorflow/python/eager:test",
|
||||
],
|
||||
)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user