Removal of BAZEL build files from python package and changes to make cpp tests work (#3641)

Co-authored-by: Marcel <maleo@google.com>

Author: apbose

BUILD.bazel

@@ -88,8 +88,7 @@ pkg_tar(
     name = "libtorchtrt",
     srcs = [
         "//:LICENSE",
-        "//bzl_def:BUILD",
-        "//bzl_def:WORKSPACE",
+        "//third_party/torch_tensorrt:BUILD",
     ],
     extension = "tar.gz",
     package_dir = "torch_tensorrt",
@@ -107,8 +106,7 @@ pkg_tar(
     name = "libtorchtrt_runtime",
     srcs = [
         "//:LICENSE",
-        "//bzl_def:BUILD",
-        "//bzl_def:WORKSPACE",
+        "//third_party/torch_tensorrt:BUILD",
     ],
     extension = "tar.gz",
     package_dir = "torch_tensorrt_runtime",

MODULE.bazel

@@ -24,14 +24,16 @@ 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",
-    path = "/opt/conda/lib/python3.11/site-packages/torch_tensorrt",
+    build_file = "@//third_party/torch_tensorrt:BUILD",
+    path = "/usr/local/lib/python3.12/site-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
 # for linux x86_64 and aarch64
 new_local_repository(

bzl_def/WORKSPACE

@@ -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/torch_tensorrt: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,29 @@ new_local_repository(
     build_file = "third_party/libtorch/BUILD"
 )
 
+
+# This is a workaround (WAR), not a functional Bazel repository.
+# It directly references a system-installed PyTorch package, which breaks hermeticity
+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"
+)

docker/MODULE.bazel.ngc

@@ -597,8 +597,6 @@ def run(self):
     package_data.update(
         {
             "torch_tensorrt": [
-                "BUILD",
-                "WORKSPACE",
                 "include/torch_tensorrt/*.h",
                 "include/torch_tensorrt/core/*.h",
                 "include/torch_tensorrt/core/conversion/*.h",
@@ -628,8 +626,6 @@ def run(self):
     package_data.update(
         {
             "torch_tensorrt": [
-                "BUILD",
-                "WORKSPACE",
                 "include/torch_tensorrt/*.h",
                 "include/torch_tensorrt/core/*.h",
                 "include/torch_tensorrt/core/runtime/*.h",

setup.py

@@ -1,6 +1,5 @@
 package(default_visibility = ["//visibility:public"])
 
 exports_files([
-    "WORKSPACE",
     "BUILD",
-])
+])