feat(//:libtrtorch): Ship trtorchc with the tarball

Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · May 31, 2020 · d647447 · d647447
1 parent f3370c4
commit d647447
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/BUILD b/BUILD
@@ -8,6 +8,8 @@ pkg_tar(
         "//core/conversion:include",
         "//core/conversion/conversionctx:include",
         "//core/conversion/converters:include",
+        "//core/conversion/var:include",
+        "//core/conversion/tensorcontainer:include",
         "//core/conversion/evaluators:include",
         "//core/execution:include",
         "//core/lowering:include",
@@ -35,6 +37,15 @@ pkg_tar(
 )
 
 
+pkg_tar(
+    name = "bin",
+    package_dir = "bin/",
+    srcs = [
+        "//cpp/trtorchc:trtorchc",
+    ],
+    mode = "0755",
+)
+
 
 
 pkg_tar(
@@ -46,6 +57,7 @@ pkg_tar(
     ],
     deps = [
         ":lib",
+        ":bin",
         ":include",
         ":include_core",
     ],

diff --git a/core/conversion/var/BUILD b/core/conversion/var/BUILD
@@ -30,7 +30,7 @@ load("@rules_pkg//:pkg.bzl", "pkg_tar")
 
 pkg_tar(
     name = "include",
-    package_dir = "core/conversion/arg/",
+    package_dir = "core/conversion/var/",
     srcs = [
         "Var.h",
         "Var_inl.h"