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