aboutsummaryrefslogtreecommitdiffhomepage
path: root/third_party/tensorrt
diff options
context:
space:
mode:
authorGravatar Sami Kama <skama@nvidia.com>2018-01-29 15:08:26 -0800
committerGravatar Sami Kama <skama@nvidia.com>2018-01-29 15:08:26 -0800
commitbe50e66950a4534322f55fbcbc300ea710f2b6c6 (patch)
treee3a70def27b622a97e3b706cb5643371d06b5fff /third_party/tensorrt
parentae740a67bdc01b991ead6ac047c774bff4d7bc8f (diff)
Fix compilation issues with new build files
Diffstat (limited to 'third_party/tensorrt')
-rw-r--r--third_party/tensorrt/BUILD.tpl31
-rw-r--r--third_party/tensorrt/build_defs.bzl.tpl1
-rw-r--r--third_party/tensorrt/tensorrt_configure.bzl4
3 files changed, 2 insertions, 34 deletions
diff --git a/third_party/tensorrt/BUILD.tpl b/third_party/tensorrt/BUILD.tpl
index 99c0e89498..dc7fe0c8c8 100644
--- a/third_party/tensorrt/BUILD.tpl
+++ b/third_party/tensorrt/BUILD.tpl
@@ -34,37 +34,6 @@ cc_library(
visibility = ["//visibility:public"],
)
-cc_library(
- name = "nv_infer_plugin",
- srcs = [%{nv_infer_plugin}],
- data = [%{nv_infer_plugin}],
- includes = [
- "include",
- ],
- copts= cuda_default_copts(),
- deps = [
- "@local_config_cuda//cuda:cuda",
- ":nv_infer",
- ":tensorrt_headers",
- ],
- linkstatic = 1,
- visibility = ["//visibility:public"],
-)
-
-cc_library(
- name = "nv_parsers",
- srcs = [%{nv_parsers}],
- data = [%{nv_parsers}],
- includes = [
- "include",
- ],
- copts= cuda_default_copts(),
- deps = [
- ":tensorrt_headers",
- ],
- linkstatic = 1,
- visibility = ["//visibility:public"],
-)
%{tensorrt_genrules}
diff --git a/third_party/tensorrt/build_defs.bzl.tpl b/third_party/tensorrt/build_defs.bzl.tpl
index f5348a7c06..0dc3a7ba2d 100644
--- a/third_party/tensorrt/build_defs.bzl.tpl
+++ b/third_party/tensorrt/build_defs.bzl.tpl
@@ -5,4 +5,3 @@ def if_tensorrt(if_true, if_false=[]):
if %{tensorrt_is_configured}:
return if_true
return if_false
-
diff --git a/third_party/tensorrt/tensorrt_configure.bzl b/third_party/tensorrt/tensorrt_configure.bzl
index 8aa0f28f39..4a1441500a 100644
--- a/third_party/tensorrt/tensorrt_configure.bzl
+++ b/third_party/tensorrt/tensorrt_configure.bzl
@@ -19,9 +19,9 @@ load(
_TENSORRT_INSTALL_PATH = "TENSORRT_INSTALL_PATH"
_TF_TENSORRT_VERSION = "TF_TENSORRT_VERSION"
-_TF_TENSORRT_LIBS = ["nvinfer", "nvinfer_plugin", "nvparsers"]
+_TF_TENSORRT_LIBS = ["nvinfer"]
_TF_TENSORRT_HEADERS = [
- "NvInfer.h", "NvInferPlugin.h", "NvCaffeParser.h", "NvUffParser.h",
+ "NvInfer.h",
"NvUtils.h"
]