44 lines
935 B
Smarty
44 lines
935 B
Smarty
|
# NVIDIA TensorRT
|
||
|
# A high-performance deep learning inference optimizer and runtime.
|
||
|
|
||
|
licenses(["notice"])
|
||
|
|
||
|
load("@local_config_cuda//cuda:build_defs.bzl", "cuda_default_copts")
|
||
|
load("@bazel_skylib//:bzl_library.bzl", "bzl_library")
|
||
|
|
||
|
package(default_visibility = ["//visibility:public"])
|
||
|
|
||
|
exports_files(["LICENSE"])
|
||
|
|
||
|
cc_library(
|
||
|
name = "tensorrt_headers",
|
||
|
hdrs = [
|
||
|
"tensorrt/include/tensorrt_config.h",
|
||
|
":tensorrt_include"
|
||
|
],
|
||
|
include_prefix = "third_party/tensorrt",
|
||
|
strip_include_prefix = "tensorrt/include",
|
||
|
)
|
||
|
|
||
|
cc_library(
|
||
|
name = "tensorrt",
|
||
|
srcs = [":tensorrt_lib"],
|
||
|
copts = cuda_default_copts(),
|
||
|
data = [":tensorrt_lib"],
|
||
|
linkstatic = 1,
|
||
|
deps = [
|
||
|
":tensorrt_headers",
|
||
|
"@local_config_cuda//cuda",
|
||
|
],
|
||
|
)
|
||
|
|
||
|
bzl_library(
|
||
|
name = "build_defs_bzl",
|
||
|
srcs = ["build_defs.bzl"],
|
||
|
deps = [
|
||
|
"@bazel_skylib//lib:selects",
|
||
|
],
|
||
|
)
|
||
|
|
||
|
%{copy_rules}
|