Merge pull request #465 from NVIDIA/headers_fix

chore(BUILD): Fix packaging of headers
pytorch · May 14, 2021 · 730efdd · 730efdd
2 parents f60bf6c + 36c1c7e
commit 730efdd
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/BUILD b/BUILD
@@ -18,9 +18,12 @@ pkg_tar(
         "//core/conversion/var:include",
         "//core/conversion/tensorcontainer:include",
         "//core/conversion/evaluators:include",
-        "//core/plugins:include",
+        "//core/ir:include",
         "//core/lowering:include",
         "//core/lowering/passes:include",
+        "//core/partitioning:include",
+        "//core/plugins:impl_include",
+        "//core/plugins:include",
         "//core/runtime:include",
         "//core/util:include",
         "//core/util/logging:include",

diff --git a/core/plugins/BUILD b/core/plugins/BUILD
@@ -42,7 +42,12 @@ load("@rules_pkg//:pkg.bzl", "pkg_tar")
 pkg_tar(
     name = "include",
     package_dir = "core/plugins/",
-    srcs = ["plugins.h",
-            "impl/interpolate_plugin.h",
-            "impl/normalize_plugin.h"],
+    srcs = ["plugins.h"],
 )
+
+pkg_tar(
+    name = "impl_include",
+    package_dir = "core/plugins/impl",
+    srcs = ["impl/interpolate_plugin.h",
+            "impl/normalize_plugin.h"],
+)