From 2c0441c286349cb75def7669c23aab79259205dc Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Thu, 14 Mar 2019 04:08:35 -0700 Subject: [PATCH] Enable all currently passing tests. I ran each test 200 times, and these passed without any flakes. PiperOrigin-RevId: 238414725 --- tensorflow/python/BUILD | 34 +++++++++++++++++++ .../data/experimental/kernel_tests/BUILD | 3 ++ tensorflow/python/data/kernel_tests/BUILD | 4 +++ tensorflow/python/distribute/BUILD | 2 ++ tensorflow/python/eager/BUILD | 10 ++++++ tensorflow/python/keras/BUILD | 6 ++++ tensorflow/python/keras/optimizer_v2/BUILD | 8 +++++ tensorflow/python/kernel_tests/BUILD | 3 ++ tensorflow/python/ops/parallel_for/BUILD | 1 + tensorflow/python/training/saving/BUILD | 1 + 10 files changed, 72 insertions(+) diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index e4e86e07065..818ee60fb43 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -1022,6 +1022,7 @@ cuda_py_test( ":platform", ":util", ], + xla_enable_strict_auto_jit = True, ) py_library( @@ -2990,6 +2991,7 @@ cuda_py_test( ":logging_ops", ":random_ops_gen", ], + xla_enable_strict_auto_jit = True, ) py_library( @@ -3513,6 +3515,7 @@ cuda_py_test( ":while_v2", "//tensorflow/python/eager:def_function", ], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -3529,6 +3532,7 @@ cuda_py_test( ":platform", "//third_party/py/numpy", ], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -3545,6 +3549,7 @@ cuda_py_test( ":platform", "//third_party/py/numpy", ], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -3576,6 +3581,7 @@ cuda_py_test( "//third_party/py/numpy", ], tags = ["no_oss"], # b/118709825 + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -3591,6 +3597,7 @@ cuda_py_test( ":variables", "//third_party/py/numpy", ], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -3645,6 +3652,7 @@ cuda_py_test( "//third_party/py/numpy", "//tensorflow/python/eager:context", ], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -3660,6 +3668,7 @@ cuda_py_test( "//third_party/py/numpy", "//tensorflow/python/eager:context", ], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -3677,6 +3686,7 @@ cuda_py_test( "//third_party/py/numpy", ], tags = ["no_windows_gpu"], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -3695,6 +3705,7 @@ cuda_py_test( "//third_party/py/numpy", ], tags = ["no_windows_gpu"], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -3767,6 +3778,7 @@ cuda_py_test( ":nn_grad", "//third_party/py/numpy", ], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -4249,6 +4261,7 @@ cuda_py_tests( ":platform_test", "//tensorflow/core:protos_all_py", ], + xla_enable_strict_auto_jit = True, ) cc_library( @@ -4627,6 +4640,7 @@ cuda_py_test( "no_oss", # Test flaky due to port collisions. "oss_serial", ], + xla_enable_strict_auto_jit = True, ) tf_py_test( @@ -4803,6 +4817,7 @@ cuda_py_test( "no_gpu", # b/127386241 "no_windows_gpu", ], + xla_enable_strict_auto_jit = True, ) tf_py_test( @@ -4893,6 +4908,7 @@ cuda_py_test( ":client_testlib", "//third_party/py/numpy", ], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -4913,6 +4929,7 @@ cuda_py_test( ":variables", ], tags = ["notsan"], + xla_enable_strict_auto_jit = True, ) cuda_py_tests( @@ -4975,6 +4992,7 @@ cuda_py_tests( "@six_archive//:six", "//tensorflow/core:protos_all_py", ], + xla_enable_strict_auto_jit = True, ) py_library( @@ -5022,6 +5040,7 @@ cuda_py_test( "//tensorflow/python/data/ops:dataset_ops", ], tags = ["multi_gpu"], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -5058,6 +5077,7 @@ cuda_py_test( "//tensorflow/core:protos_all_py", "//tensorflow/python/data/ops:dataset_ops", ], + xla_enable_strict_auto_jit = True, ) tf_py_test( @@ -5115,6 +5135,7 @@ cuda_py_test( ], grpc_enabled = True, main = "training/session_manager_test.py", + xla_enable_strict_auto_jit = True, ) tf_py_test( @@ -5612,6 +5633,7 @@ cuda_py_test( ], main = "ops/accumulate_n_benchmark.py", shard_count = 6, + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -5632,6 +5654,7 @@ cuda_py_test( ":variables", ], main = "ops/batch_norm_benchmark.py", + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -5649,6 +5672,7 @@ cuda_py_test( "//tensorflow/core:protos_all_py", ], main = "ops/concat_benchmark.py", + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -5662,6 +5686,7 @@ cuda_py_test( "//tensorflow/python/eager:function", ], main = "ops/control_flow_ops_benchmark.py", + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -5682,6 +5707,7 @@ cuda_py_test( "//tensorflow/core:protos_all_py", ], main = "ops/conv2d_benchmark.py", + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -5700,6 +5726,7 @@ cuda_py_test( "//tensorflow/core:protos_all_py", ], main = "ops/split_benchmark.py", + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -5719,6 +5746,7 @@ cuda_py_test( "//tensorflow/core:protos_all_py", ], main = "ops/transpose_benchmark.py", + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -5740,6 +5768,7 @@ cuda_py_test( "//tensorflow/core:protos_all_py", ], main = "ops/matmul_benchmark.py", + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -5762,6 +5791,7 @@ cuda_py_test( ], main = "ops/matmul_benchmark_test.py", tags = ["no_pip"], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -5779,6 +5809,7 @@ cuda_py_test( ], grpc_enabled = True, main = "client/session_benchmark.py", + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -5792,6 +5823,7 @@ cuda_py_test( ":nn_ops", "//third_party/py/numpy", ], + xla_enable_strict_auto_jit = True, ) py_library( @@ -5989,6 +6021,7 @@ cuda_py_test( tags = [ "grappler", ], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -6181,4 +6214,5 @@ cuda_py_test( additional_deps = [ "//tensorflow/python:client_testlib", ], + xla_enable_strict_auto_jit = True, ) diff --git a/tensorflow/python/data/experimental/kernel_tests/BUILD b/tensorflow/python/data/experimental/kernel_tests/BUILD index 20b8bb9f2cd..50626e48178 100644 --- a/tensorflow/python/data/experimental/kernel_tests/BUILD +++ b/tensorflow/python/data/experimental/kernel_tests/BUILD @@ -45,6 +45,7 @@ cuda_py_test( "//tensorflow/python/data/ops:iterator_ops", ], tags = ["no_windows_gpu"], + xla_enable_strict_auto_jit = True, ) py_test( @@ -461,6 +462,7 @@ cuda_py_test( "//tensorflow/python/data/ops:dataset_ops", ], tags = ["no_windows_gpu"], + xla_enable_strict_auto_jit = True, ) py_library( @@ -759,4 +761,5 @@ cuda_py_test( "//tensorflow/python/data/kernel_tests:test_base", "//tensorflow/python/data/ops:dataset_ops", ], + xla_enable_strict_auto_jit = True, ) diff --git a/tensorflow/python/data/kernel_tests/BUILD b/tensorflow/python/data/kernel_tests/BUILD index d31877f2049..b56049f32da 100644 --- a/tensorflow/python/data/kernel_tests/BUILD +++ b/tensorflow/python/data/kernel_tests/BUILD @@ -370,6 +370,7 @@ cuda_py_test( "//tensorflow/python:variables", ], grpc_enabled = True, + xla_enable_strict_auto_jit = True, ) tf_py_test( @@ -442,6 +443,7 @@ cuda_py_test( "no_oss", # TODO(b/117920141): Investigate breakage and re-enable. "no_windows_gpu", ], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -457,6 +459,7 @@ cuda_py_test( "//tensorflow/python:client_testlib", "//tensorflow/python:framework_test_lib", ], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -478,6 +481,7 @@ cuda_py_test( "//tensorflow/python:framework_test_lib", "//tensorflow/python:tensor_shape", ], + xla_enable_strict_auto_jit = True, ) tf_py_test( diff --git a/tensorflow/python/distribute/BUILD b/tensorflow/python/distribute/BUILD index 62c40eacfbd..308bd428d05 100644 --- a/tensorflow/python/distribute/BUILD +++ b/tensorflow/python/distribute/BUILD @@ -102,6 +102,7 @@ cuda_py_test( "//tensorflow/python:client_testlib", "//tensorflow/python:framework_ops", ], + xla_enable_strict_auto_jit = True, ) py_library( @@ -387,6 +388,7 @@ cuda_py_test( "//tensorflow/python:io_ops", "//tensorflow/python:util", ], + xla_enable_strict_auto_jit = True, ) py_test( diff --git a/tensorflow/python/eager/BUILD b/tensorflow/python/eager/BUILD index f1f26035887..0c3db5114d7 100644 --- a/tensorflow/python/eager/BUILD +++ b/tensorflow/python/eager/BUILD @@ -115,6 +115,7 @@ cuda_py_test( "//tensorflow/python:constant_op", "//tensorflow/core/profiler:protos_all_py", ], + xla_enable_strict_auto_jit = True, ) py_library( @@ -143,6 +144,7 @@ cuda_py_test( "//tensorflow/python:errors", "//tensorflow/python:framework_ops", ], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -182,6 +184,7 @@ cuda_py_test( "//tensorflow/python:framework_test_lib", "//tensorflow/python:pywrap_tensorflow", ], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -196,6 +199,7 @@ cuda_py_test( "@absl_py//absl/testing:parameterized", "//tensorflow/python:math_ops", ], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -211,6 +215,7 @@ cuda_py_test( "//tensorflow/python:math_ops", "//tensorflow/python:resource_variable_ops", ], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -228,6 +233,7 @@ cuda_py_test( "//tensorflow/python:resource_variable_ops", ], shard_count = 5, + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -328,6 +334,7 @@ cuda_py_test( "//tensorflow/python:framework_test_lib", "//tensorflow/python:math_ops", ], + xla_enable_strict_auto_jit = True, ) py_library( @@ -405,6 +412,7 @@ cuda_py_test( "//tensorflow/python:random_ops", "//tensorflow/python/keras", ], + xla_enable_strict_auto_jit = True, ) tf_py_logged_benchmark( @@ -451,6 +459,7 @@ cuda_py_test( "//tensorflow/python:sparse_ops", "//tensorflow/python:tensor_shape", ], + xla_enable_strict_auto_jit = True, ) tf_py_test( @@ -496,6 +505,7 @@ cuda_py_test( tags = [ "optonly", # The test is too slow in non-opt mode ], + xla_enable_strict_auto_jit = True, ) py_library( diff --git a/tensorflow/python/keras/BUILD b/tensorflow/python/keras/BUILD index e8782951062..ace1716f1f3 100755 --- a/tensorflow/python/keras/BUILD +++ b/tensorflow/python/keras/BUILD @@ -606,6 +606,7 @@ cuda_py_test( "//third_party/py/numpy", "//tensorflow/python:client_testlib", ], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -623,6 +624,7 @@ cuda_py_test( "no_rocm", "no_windows_gpu", ], + xla_enable_strict_auto_jit = True, ) tf_py_test( @@ -675,6 +677,7 @@ cuda_py_test( "@absl_py//absl/testing:parameterized", "//tensorflow/python:client_testlib", ], + xla_enable_strict_auto_jit = True, ) tf_py_test( @@ -809,6 +812,7 @@ cuda_py_test( "//third_party/py/numpy", "//tensorflow/python:client_testlib", ], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -1036,6 +1040,7 @@ cuda_py_test( "guitar", "multi_gpu", ], + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -1048,6 +1053,7 @@ cuda_py_test( "//third_party/py/numpy", "//tensorflow/python:client_testlib", ], + xla_enable_strict_auto_jit = True, ) tf_py_test( diff --git a/tensorflow/python/keras/optimizer_v2/BUILD b/tensorflow/python/keras/optimizer_v2/BUILD index f521a056b69..e15cf80466b 100644 --- a/tensorflow/python/keras/optimizer_v2/BUILD +++ b/tensorflow/python/keras/optimizer_v2/BUILD @@ -72,6 +72,7 @@ cuda_py_test( "//tensorflow/python/eager:context", ], shard_count = 4, + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -91,6 +92,7 @@ cuda_py_test( "//tensorflow/python/eager:context", ], shard_count = 4, + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -110,6 +112,7 @@ cuda_py_test( "//tensorflow/python/eager:context", ], shard_count = 4, + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -129,6 +132,7 @@ cuda_py_test( "//tensorflow/python/eager:context", ], shard_count = 4, + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -148,6 +152,7 @@ cuda_py_test( "//tensorflow/python/eager:context", ], shard_count = 4, + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -167,6 +172,7 @@ cuda_py_test( "//tensorflow/python/eager:context", ], shard_count = 4, + xla_enable_strict_auto_jit = True, ) cuda_py_test( @@ -186,6 +192,7 @@ cuda_py_test( "//tensorflow/python/eager:context", ], shard_count = 4, + xla_enable_strict_auto_jit = True, ) py_test( @@ -249,4 +256,5 @@ cuda_py_test( "//tensorflow/python:variables", ], shard_count = 2, + xla_enable_strict_auto_jit = True, ) diff --git a/tensorflow/python/kernel_tests/BUILD b/tensorflow/python/kernel_tests/BUILD index 8b7650d95da..71c1a1e2df1 100644 --- a/tensorflow/python/kernel_tests/BUILD +++ b/tensorflow/python/kernel_tests/BUILD @@ -308,6 +308,7 @@ cuda_py_test( "//tensorflow/python:client_testlib", "//tensorflow/python:nn_ops", ], + xla_enable_strict_auto_jit = True, ) tf_py_test( @@ -1194,6 +1195,7 @@ cuda_py_test( "//tensorflow/python:variable_scope", "//tensorflow/python:variables", ], + xla_enable_strict_auto_jit = True, ) tf_py_test( @@ -3147,6 +3149,7 @@ cuda_py_test( "no_pip", "notap", # http://b/31080670 ], + xla_enable_strict_auto_jit = True, ) cuda_py_test( diff --git a/tensorflow/python/ops/parallel_for/BUILD b/tensorflow/python/ops/parallel_for/BUILD index 0a2f3e25a29..0dd13a420f6 100644 --- a/tensorflow/python/ops/parallel_for/BUILD +++ b/tensorflow/python/ops/parallel_for/BUILD @@ -116,6 +116,7 @@ cuda_py_test( "//tensorflow/python:util", ], tags = ["no_rocm"], + xla_enable_strict_auto_jit = True, ) cuda_py_test( diff --git a/tensorflow/python/training/saving/BUILD b/tensorflow/python/training/saving/BUILD index adb50f9aa6b..5577a2b4ae8 100644 --- a/tensorflow/python/training/saving/BUILD +++ b/tensorflow/python/training/saving/BUILD @@ -34,6 +34,7 @@ cuda_py_test( ":functional_saver", "//tensorflow/python/eager:test", ], + xla_enable_strict_auto_jit = True, ) py_library(