diff options
| author |  Gunhan Gulsoy <gunan@google.com> | 2018-09-14 14:07:14 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-14 14:10:58 -0700 |
| commit | 19d66a950e2091bb598c6a2d375e14208f5773b2 (patch) | |
| tree | 61b67f566aca38534f58302fee227c26818cc52b /tensorflow/contrib/tensorrt | |
| parent | f104b477ab22d5bc71afa757ec0cdeaca8666909 (diff) | |
Disable flaky gpu_base_test
PiperOrigin-RevId: 213040362
Diffstat (limited to 'tensorflow/contrib/tensorrt')
| -rw-r--r-- | tensorflow/contrib/tensorrt/BUILD | 21 |
1 files changed, 20 insertions, 1 deletions
diff --git a/tensorflow/contrib/tensorrt/BUILD b/tensorflow/contrib/tensorrt/BUILD index 9e8979bce4..4ea7216ef2 100644 --- 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"], |