github.com/johnnyeven/libtools@v0.0.0-20191126065708-61829c1adf46/third_party/toolchains/preconfig/ubuntu16.04/tensorrt5.1/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 = [":tensorrt_include"],
    15      include_prefix = "third_party/tensorrt",
    16      strip_include_prefix = "tensorrt/include",
    17  )
    18  
    19  cc_library(
    20      name = "tensorrt",
    21      srcs = [":tensorrt_lib"],
    22      copts = cuda_default_copts(),
    23      data = [":tensorrt_lib"],
    24      linkstatic = 1,
    25      deps = [
    26          ":tensorrt_headers",
    27          "@local_config_cuda//cuda",
    28      ],
    29  )
    30  
    31  genrule(
    32      name = "tensorrt_lib",
    33      outs = [
    34          "tensorrt/lib/libnvinfer.so.5",
    35          "tensorrt/lib/libnvinfer_plugin.so.5",
    36      ],
    37      cmd = """cp -f "/usr/local/tensorrt/lib/libnvinfer.so.5" "$(location tensorrt/lib/libnvinfer.so.5)" && \
    38  cp -f "/usr/local/tensorrt/lib/libnvinfer_plugin.so.5" "$(location tensorrt/lib/libnvinfer_plugin.so.5)" """,
    39  )
    40  
    41  genrule(
    42      name = "tensorrt_include",
    43      outs = [
    44          "tensorrt/include/NvInfer.h",
    45          "tensorrt/include/NvUtils.h",
    46          "tensorrt/include/NvInferPlugin.h",
    47      ],
    48      cmd = """cp -f "/usr/local/tensorrt/include/NvInfer.h" "$(location tensorrt/include/NvInfer.h)" && \
    49  cp -f "/usr/local/tensorrt/include/NvUtils.h" "$(location tensorrt/include/NvUtils.h)" && \
    50  cp -f "/usr/local/tensorrt/include/NvInferPlugin.h" "$(location tensorrt/include/NvInferPlugin.h)" """,
    51  )