diff options
author | Guangda Lai <laigd@google.com> | 2018-07-16 22:56:37 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-16 23:01:37 -0700 |
commit | c091185930eee5a3f87cfbe5e367a3cfc8b717e8 (patch) | |
tree | da95c09c4bf1cae335e6dc64735fd993c63ed625 /tensorflow/contrib/BUILD | |
parent | a013474eae601ae27c406e6c8da34ef762e8f762 (diff) |
Fix various building/testing issues and enable basic TensorRT tests, details are:
1. Add IsGoogleTensorRTEnabled() method and is_tensorrt_enabled() python wrapper to guard trt python tests.
2. Fix nvcc build problems and add corresponding TODOs in some c++ code.
3. Fix various kokoro test config problems (e.g. fix oss build dependencies, add nomac tags for some tests, etc)
PiperOrigin-RevId: 204862004
Diffstat (limited to 'tensorflow/contrib/BUILD')
-rw-r--r-- | tensorflow/contrib/BUILD | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/tensorflow/contrib/BUILD b/tensorflow/contrib/BUILD index 60be9db263..1322056d80 100644 --- a/tensorflow/contrib/BUILD +++ b/tensorflow/contrib/BUILD @@ -7,7 +7,6 @@ package(default_visibility = ["//tensorflow:__subpackages__"]) load("//third_party/mpi:mpi.bzl", "if_mpi") load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda") -load("@local_config_tensorrt//:build_defs.bzl", "if_tensorrt") load("//tensorflow:tensorflow.bzl", "if_not_windows") load("//tensorflow:tensorflow.bzl", "if_not_windows_cuda") @@ -103,6 +102,7 @@ py_library( "//tensorflow/contrib/summary:summary", "//tensorflow/contrib/tensor_forest:init_py", "//tensorflow/contrib/tensorboard", + "//tensorflow/contrib/tensorrt:init_py", "//tensorflow/contrib/testing:testing_py", "//tensorflow/contrib/text:text_py", "//tensorflow/contrib/tfprof", @@ -113,9 +113,7 @@ py_library( "//tensorflow/contrib/util:util_py", "//tensorflow/python:util", "//tensorflow/python/estimator:estimator_py", - ] + if_mpi(["//tensorflow/contrib/mpi_collectives:mpi_collectives_py"]) + if_tensorrt([ - "//tensorflow/contrib/tensorrt:init_py", - ]) + select({ + ] + if_mpi(["//tensorflow/contrib/mpi_collectives:mpi_collectives_py"]) + select({ "//tensorflow:with_kafka_support_windows_override": [], "//tensorflow:with_kafka_support": [ "//tensorflow/contrib/kafka", |