Automated rollback of commit d78595d333

PiperOrigin-RevId: 215073584

tensorflow/python/data/kernel_tests/BUILD

@@ -318,7 +318,6 @@ cuda_py_test(
         "//tensorflow/python:framework_test_lib",
     ],
     tags = [
-        "no_oss",  # TODO(b/116813115): Investigate timeout and re-enable.
         "no_windows_gpu",
     ],
 )

tensorflow/tensorflow.bzl

@@ -1798,29 +1798,22 @@ def cuda_py_test(
         flaky = 0,
         xla_enabled = False,
         grpc_enabled = False):
-    if main == None:
+    test_tags = tags + tf_cuda_tests_tags()
-        main = name + ".py"
+    tf_py_test(
-    for config in ["cpu", "gpu"]:
+        name = name,
-        test_name = name
+        size = size,
-        test_tags = tags
+        srcs = srcs,
-        if config == "gpu":
+        data = data,
-            test_name += "_gpu"
+        main = main,
-            test_tags = test_tags + tf_cuda_tests_tags()
+        args = args,
-        tf_py_test(
+        tags = test_tags,
-            name = test_name,
+        shard_count = shard_count,
-            size = size,
+        additional_deps = additional_deps,
-            srcs = srcs,
+        kernels = kernels,
-            data = data,
+        flaky = flaky,
-            main = main,
+        xla_enabled = xla_enabled,
-            args = args,
+        grpc_enabled = grpc_enabled,
-            tags = test_tags,
+    )
-            shard_count = shard_count,
-            additional_deps = additional_deps,
-            kernels = kernels,
-            flaky = flaky,
-            xla_enabled = xla_enabled,
-            grpc_enabled = grpc_enabled,
-        )
 
 register_extension_info(
     extension_name = "cuda_py_test",

tensorflow/tools/pip_package/pip_smoke_test.py

@@ -142,7 +142,7 @@ def main():
 
   missing_dependencies = []
   # File extensions and endings to ignore
-  ignore_extensions = ["_test", "_test.py", "_test_gpu", "_test_gpu.py"]
+  ignore_extensions = ["_test", "_test.py"]
 
   ignored_files = 0
   blacklisted_files = len(BLACKLIST)