github.com/johnnyeven/libtools@v0.0.0-20191126065708-61829c1adf46/third_party/toolchains/preconfig/ubuntu16.04/tensorrt5/BUILD (about)

     1  # NVIDIA TensorRT
     2  # A high-performance deep learning inference optimizer and runtime.
     3  
     4  licenses(["notice"])
     5  
     6  load("@local_config_cuda//cuda:build_defs.bzl", "cuda_default_copts")
     7  
     8  package(default_visibility = ["//visibility:public"])
     9  
    10  exports_files(["LICENSE"])
    11  
    12  cc_library(
    13      name = "tensorrt_headers",
    14      hdrs = [
    15          "tensorrt/include/tensorrt_config.h",
    16          ":tensorrt_include",
    17      ],
    18      include_prefix = "third_party/tensorrt",
    19      strip_include_prefix = "tensorrt/include",
    20  )
    21  
    22  cc_library(
    23      name = "tensorrt",
    24      srcs = [":tensorrt_lib"],
    25      copts = cuda_default_copts(),
    26      data = [":tensorrt_lib"],
    27      linkstatic = 1,
    28      deps = [
    29          ":tensorrt_headers",
    30          "@local_config_cuda//cuda",
    31      ],
    32  )
    33  
    34  genrule(
    35      name = "tensorrt_lib",
    36      outs = [
    37          "tensorrt/lib/libnvinfer.so.5",
    38          "tensorrt/lib/libnvinfer_plugin.so.5",
    39      ],
    40      cmd = """cp -f "/usr/lib/x86_64-linux-gnu/libnvinfer.so.5" "$(location tensorrt/lib/libnvinfer.so.5)" && \
    41  cp -f "/usr/lib/x86_64-linux-gnu/libnvinfer_plugin.so.5" "$(location tensorrt/lib/libnvinfer_plugin.so.5)" """,
    42  )
    43  
    44  genrule(
    45      name = "tensorrt_include",
    46      outs = [
    47          "tensorrt/include/NvInfer.h",
    48          "tensorrt/include/NvUtils.h",
    49          "tensorrt/include/NvInferPlugin.h",
    50      ],
    51      cmd = """cp -f "/usr/include/x86_64-linux-gnu/NvInfer.h" "$(location tensorrt/include/NvInfer.h)" && \
    52  cp -f "/usr/include/x86_64-linux-gnu/NvUtils.h" "$(location tensorrt/include/NvUtils.h)" && \
    53  cp -f "/usr/include/x86_64-linux-gnu/NvInferPlugin.h" "$(location tensorrt/include/NvInferPlugin.h)" """,
    54  )