rules_tensorrt_rtx/BUILD.bazel at main · wep21/rules_tensorrt_rtx · GitHub

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
load("@rules_cc//cc:defs.bzl", "cc_library")
load("@rules_cuda//cuda:defs.bzl", "requires_cuda")

cc_library(
    name = "tensorrt_rtx",
    target_compatible_with = requires_cuda(),
    visibility = ["//visibility:public"],
    deps = select({
        "@platforms//os:linux": ["@tensorrt_rtx_linux_x86_64//:tensorrt_rtx"],
        "@platforms//os:windows": ["@tensorrt_rtx_windows_x86_64//:tensorrt_rtx"],
    }),
)

cc_library(
    name = "tensorrt_onnxparser_rtx",
    target_compatible_with = requires_cuda(),
    visibility = ["//visibility:public"],
    deps = select({
        "@platforms//os:linux": ["@tensorrt_rtx_linux_x86_64//:tensorrt_onnxparser_rtx"],
        "@platforms//os:windows": ["@tensorrt_rtx_windows_x86_64//:tensorrt_onnxparser_rtx"],
    }),
)

alias(
    name = "tensorrt_rtx_bin",
    actual = select({
        "@platforms//os:linux": "@tensorrt_rtx_linux_x86_64//:tensorrt_rtx_bin",
        "@platforms//os:windows": "@tensorrt_rtx_windows_x86_64//:tensorrt_rtx_bin",
    }),
    target_compatible_with = requires_cuda(),
)