Disable flaky gpu_base_test

PiperOrigin-RevId: 213040362
ganny26 · Sep 14, 2018 · 19d66a9 · 19d66a9
1 parent f104b47
commit 19d66a9
Showing 1 changed file with 20 additions and 1 deletion.
diff --git a/tensorflow/contrib/tensorrt/BUILD b/tensorflow/contrib/tensorrt/BUILD
@@ -444,7 +444,6 @@ cuda_py_test(
 cuda_py_tests(
     name = "tf_trt_integration_test",
     srcs = [
-        "test/base_test.py",
         "test/batch_matmul_test.py",
         "test/biasadd_matmul_test.py",
         "test/binary_tensor_weight_broadcast_test.py",
@@ -471,6 +470,26 @@ cuda_py_tests(
     ],
 )
 
+cuda_py_tests(
+    name = "base_test",
+    srcs = [
+        "test/base_test.py",
+    ],
+    additional_deps = [
+        ":tf_trt_integration_test_base",
+        "//tensorflow/python:client_testlib",
+        "//tensorflow/python:framework_test_lib",
+    ],
+    tags = [
+        "manual",
+        "no_cuda_on_cpu_tap",
+        "no_gpu",
+        "no_windows",
+        "nomac",
+        "notap",
+    ],
+)
+
 cc_library(
     name = "utils",
     srcs = ["convert/utils.cc"],