1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
# NVIDIA TensorRT Open Source Plugins
# This package contains build targets for select TensorRT plugins included in the
# TensorRT open source repository.
load("@local_config_cuda//cuda:build_defs.bzl", "cuda_default_copts", "cuda_library")
exports_files(["LICENSE"])
cuda_library(
name = "plugin_common",
srcs = [
"plugin/common/kernels/common.cu.cc",
],
hdrs = [
"plugin/common/bboxUtils.h",
"plugin/common/checkMacrosPlugin.h",
"plugin/common/plugin.h",
],
strip_include_prefix = "plugin/common",
deps = [
"@local_config_tensorrt//:tensorrt",
"@local_config_tensorrt//:tensorrt_headers",
],
)
cc_library(
name = "nms_plugin_hdrs",
hdrs = [
"plugin/efficientNMSPlugin/efficientNMSInference.h",
"plugin/efficientNMSPlugin/efficientNMSParameters.h",
"plugin/efficientNMSPlugin/efficientNMSPlugin.h",
],
)
cuda_library(
name = "nvinfer_plugin_nms",
srcs = [
"plugin/efficientNMSPlugin/efficientNMSInference.cu.cc",
"plugin/efficientNMSPlugin/efficientNMSInference.cu.h",
"plugin/efficientNMSPlugin/efficientNMSInference.h",
"plugin/efficientNMSPlugin/efficientNMSParameters.h",
"plugin/efficientNMSPlugin/efficientNMSPlugin.cpp",
"plugin/efficientNMSPlugin/efficientNMSPlugin.h",
],
hdrs = [
"plugin/efficientNMSPlugin/efficientNMSInference.h",
"plugin/efficientNMSPlugin/efficientNMSParameters.h",
"plugin/efficientNMSPlugin/efficientNMSPlugin.h",
],
copts = cuda_default_copts(),
include_prefix = "third_party/tensorrt/plugin/efficientNMSPlugin",
strip_include_prefix = "plugin/efficientNMSPlugin",
visibility = ["//visibility:public"],
deps = [
":nms_plugin_hdrs",
":plugin_common",
"@local_config_tensorrt//:tensorrt",
"@local_config_tensorrt//:tensorrt_headers",
],
)
|
