chore: example on how to use external trtorch as a dep

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

BUILD

@@ -65,7 +65,7 @@ pkg_tar(
     name = "libtrtorch",
     srcs = [
         "//:LICENSE",
-        "//bzl_def:BUILD.bzl",
+        "//bzl_def:BUILD",
         "//bzl_def:WORKSPACE"
     ],
     extension = "tar.gz",

bzl_def/BUILD

@@ -1,6 +1,79 @@
 package(default_visibility = ["//visibility:public"])
 
-exports_files([
-	"WORKSPACE",
-	"BUILD.bzl"
-])
+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 = "libtrtorch",
+    srcs = select({
+    ":windows": [
+        "lib/x64/trtorch.dll",
+    ],
+    "//conditions:default": [
+        "lib/libtrtorch.so",
+    ],
+    }),
+    hdrs = glob([
+        "include/**/*.h",
+    ]),
+    strip_include_prefix = "include",
+    includes = ["include/"]
+)
+
+cc_library(
+    name = "libtrtorchrt",
+    srcs = select({
+        ":windows": [
+            "lib/x64/trtorchrt.dll"
+        ],
+        "//conditions:default": [
+            "lib/libtrtorchrt.so"
+        ]
+    })
+)
+
+cc_library(
+    name = "libtrtorch_plugins",
+    srcs = select({
+        ":windows": [
+            "lib/x64/trtorch_plugins.dll"
+        ],
+        "//conditions:default": [
+            "lib/libtrtorch_plugins.so"
+        ]
+    }),
+    hdrs = glob([
+        "include/trtorch/core/plugins/**/*.h",
+    ]),
+    strip_include_prefix = "include",
+    includes = ["include/"]
+)
+
+cc_library(
+    name = "trtorch_core_hdrs",
+    hdrs = glob([
+        "include/trtorch/core/**/*.h"
+    ]),
+    strip_include_prefix = "include/trtorch",
+    includes = ["include/trtorch/"]
+)
+
+# Alias for ease of use
+cc_library(
+    name = "trtorch",
+    deps = [
+        ":libtrtorch",
+    ]
+)

bzl_def/BUILD.bazel

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

bzl_def/BUILD.bzl

@@ -7,6 +7,13 @@ config_setting(
     },
 )
 
+config_setting(
+    name = "ci_build_testing",
+    values = {
+        "define": "trtorch_src=pre_built"
+    }
+)
+
 cc_library(
     name = "util",
     srcs = [
@@ -20,9 +27,6 @@ cc_library(
         "util.h",
     ],
     deps = [
-        "//core/conversion",
-        "//core/util:prelude",
-        "//cpp:trtorch",
         "@tensorrt//:nvinfer",
     ] + select({
         ":use_pre_cxx11_abi": [
@@ -33,5 +37,15 @@ cc_library(
             "@libtorch//:libtorch",
             "@libtorch//:caffe2",
         ],
+    }) + select({
+        ":ci_build_testing": [
+            "@trtorch//:trtorch",
+            "@trtorch//:trtorch_core_hdrs"
+        ],
+        "//conditions:default": [
+            "//cpp:trtorch",
+            "//core/conversion",
+            "//core/util:prelude"
+        ]
     }),
 )