changes for DLFW for l4t target and removal of BUILD files from package

Author: apbose

docker/MODULE.bazel.ngc

@@ -24,14 +24,17 @@ git_override(
 
 local_repository = use_repo_rule("@bazel_tools//tools/build_defs/repo:local.bzl", "local_repository")
 
+
+new_local_repository = use_repo_rule("@bazel_tools//tools/build_defs/repo:local.bzl", "new_local_repository")
+
+
 # External dependency for torch_tensorrt if you already have precompiled binaries.
-local_repository(
+new_local_repository(
     name = "torch_tensorrt",
+    build_file = "@//third_party:BUILD",
     path = "/usr/local/lib/python3.12/dist-packages/torch_tensorrt/",
 )
 
-new_local_repository = use_repo_rule("@bazel_tools//tools/build_defs/repo:local.bzl", "new_local_repository")
-
 
 # CUDA should be installed on the system locally
 new_local_repository(
@@ -55,8 +58,26 @@ new_local_repository(
     build_file = "third_party/libtorch/BUILD"
 )
 
+new_local_repository(
+    name = "torch_l4t",
+    path = "/usr/local/lib/python3.12/dist-packages/torch/",
+    build_file = "third_party/libtorch/BUILD"
+)
+
 new_local_repository(
    name = "tensorrt",
    path = "/usr/",
    build_file = "@//third_party/tensorrt/local:BUILD"
 )
+
+new_local_repository(
+   name = "tensorrt_sbsa",
+   path = "/usr/",
+   build_file = "@//third_party/tensorrt/local:BUILD"
+)
+
+new_local_repository(
+   name = "tensorrt_l4t",
+   path = "/usr/",
+   build_file = "@//third_party/tensorrt/local:BUILD"
+)

third_party/BUILD

@@ -0,0 +1,83 @@
+load("@rules_cc//cc:cc_library.bzl", "cc_library")
+
+package(default_visibility = ["//visibility:public"])
+
+load("@rules_cc//cc:defs.bzl", "cc_library")
+
+config_setting(
+    name = "aarch64_linux",
+    constraint_values = [
+        "@platforms//cpu:aarch64",
+        "@platforms//os:linux",
+    ],
+)
+
+config_setting(
+    name = "windows",
+    constraint_values = [
+        "@platforms//os:windows",
+    ],
+)
+
+cc_library(
+    name = "libtorchtrt",
+    srcs = select({
+        ":windows": [
+            "lib/x64/torchtrt.dll",
+        ],
+        "//conditions:default": [
+            "lib/libtorchtrt.so",
+        ],
+    }),
+    hdrs = glob([
+        "include/**/*.h",
+    ]),
+    includes = ["include/"],
+    strip_include_prefix = "include",
+)
+
+cc_library(
+    name = "libtorchtrt_runtime",
+    srcs = select({
+        ":windows": [
+            "lib/x64/torchtrt_runtime.dll",
+        ],
+        "//conditions:default": [
+            "lib/libtorchtrt_runtime.so",
+        ],
+    }),
+)
+
+cc_library(
+    name = "libtorchtrt_plugins",
+    srcs = select({
+        ":windows": [
+            "lib/x64/torchtrt_plugins.dll",
+        ],
+        "//conditions:default": [
+            "lib/libtorchtrt_plugins.so",
+        ],
+    }),
+    hdrs = glob([
+        "include/torch_tensorrt/core/plugins/**/*.h",
+    ]),
+    includes = ["include/"],
+    strip_include_prefix = "include",
+)
+
+cc_library(
+    name = "torch_tensorrt_core_hdrs",
+    hdrs = glob([
+        "include/torch_tensorrt/core/**/*.h",
+    ]),
+    includes = ["include/torch_tensorrt/"],
+    strip_include_prefix = "include/torch_tensorrt",
+)
+
+# Alias for ease of use
+cc_library(
+    name = "torch_tensorrt",
+    deps = [
+        ":libtorchtrt",
+    ],
+)