github.com/johnnyeven/libtools@v0.0.0-20191126065708-61829c1adf46/third_party/tensorrt/BUILD.tpl (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  %{copy_rules}