Merge pull request #465 from NVIDIA/headers_fix

chore(BUILD): Fix packaging of headers

Author: narendasan

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",

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"],
+)