diff options
author | Todd Wang <toddw@google.com> | 2018-10-04 15:29:58 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-04 15:37:01 -0700 |
commit | a08ca5bb74fcd828c19060216923ad0f378bb518 (patch) | |
tree | f206457353d5e4e28bfc3f24ef97b7ce46594084 /tensorflow/contrib/tensorrt | |
parent | 3a457c7252f09afd03483092ce9dcc7aa292b8c6 (diff) |
Disable tensorrt:unary_test in OSS since it crashes with SEGV.
PiperOrigin-RevId: 215814732
Diffstat (limited to 'tensorflow/contrib/tensorrt')
-rw-r--r-- | tensorflow/contrib/tensorrt/BUILD | 20 |
1 files changed, 19 insertions, 1 deletions
diff --git a/tensorflow/contrib/tensorrt/BUILD b/tensorflow/contrib/tensorrt/BUILD index 9e8979bce4..5c16fcb760 100644 --- a/tensorflow/contrib/tensorrt/BUILD +++ b/tensorflow/contrib/tensorrt/BUILD @@ -455,7 +455,6 @@ cuda_py_tests( "test/multi_connection_neighbor_engine_test.py", "test/neighboring_engine_test.py", "test/rank_two_test.py", - "test/unary_test.py", "test/vgg_block_nchw_test.py", "test/vgg_block_test.py", ], @@ -471,6 +470,25 @@ cuda_py_tests( ], ) +cuda_py_tests( + name = "tf_trt_integration_test_no_oss", + srcs = [ + "test/unary_test.py", + ], + additional_deps = [ + ":tf_trt_integration_test_base", + "//tensorflow/python:client_testlib", + "//tensorflow/python:framework_test_lib", + ], + tags = [ + "no_cuda_on_cpu_tap", + "no_oss", # TODO(b/117274186): re-enable in OSS after crash fixed + "no_pip", # TODO(b/117274186): re-enable in OSS after crash fixed + "no_windows", + "nomac", + ], +) + cc_library( name = "utils", srcs = ["convert/utils.cc"], |