# MODULE.bazel
bazel_dep(name = "rules_tensorrt_rtx", version = "0.1.0")
tensorrt_rtx = use_extension("@rules_tensorrt_rtx//:extensions.bzl", "tensorrt_rtx")
tensorrt_rtx.fetch(
# TensorRT RTX 1.2.0.54 defaults to CUDA 12.9 artifacts; override to use CUDA 13.0.
cuda_version = "12.9",
# version = "1.2.0.54",
)
use_repo(
tensorrt_rtx,
"tensorrt_rtx_linux_x86_64",
"tensorrt_rtx_windows_x86_64",
)
# BUILD.bazel
cc_binary(
name = "tensorrt_rtx_main",
srcs = ["main.cpp"],
deps = [
"@rules_tensorrt_rtx//:tensorrt_rtx",
"@rules_tensorrt_rtx//:tensorrt_onnxparser_rtx",
],
)
-
Notifications
You must be signed in to change notification settings - Fork 0
wep21/rules_tensorrt_rtx
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.