chore: Add PEX, SCIE binary build configs for the plugin subsystem (#…

…2422) Co-authored-by: Joongi Kim <[email protected]>
lablup · Jul 17, 2024 · 60af16b · 60af16b
1 parent 19df64c
commit 60af16b
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 1 deletion.
diff --git a/changes/2422.misc.md b/changes/2422.misc.md
@@ -0,0 +1 @@
+Add PEX, SCIE binary build configs for the plugin subsystem.
diff --git a/src/ai/backend/cli/BUILD b/src/ai/backend/cli/BUILD
@@ -14,6 +14,7 @@ visibility_private_component(
         "//src/ai/backend/web/**",
         "//src/ai/backend/install/**",
         "//src/ai/backend/wsproxy/**",
+        "//src/ai/backend/plugin/**",
     ],
     allowed_dependencies=[
         "//src/ai/backend/plugin/**",

diff --git a/src/ai/backend/plugin/BUILD b/src/ai/backend/plugin/BUILD
@@ -18,7 +18,9 @@ visibility_private_component(
         "//src/ai/backend/testutils/**",
         "//src/ai/backend/install/**",
     ],
-    allowed_dependencies=[],
+    allowed_dependencies=[
+        "//src/ai/backend/cli/**",
+    ],
 )
 
 python_distribution(
@@ -52,3 +54,29 @@ resources(
         "**/py.typed",
     ],
 )
+
+pex_binary(
+    name="pex",
+    entry_point="ai.backend.cli.__main__",
+    dependencies=[
+        ":src",
+        ":buildscript",
+        "!!stubs/trafaret:stubs",
+    ],
+)
+
+scie_binary(
+    name="backendai-plugin",
+    fat=False,
+    dependencies=[":pex"],
+    tags=["scie", "lazy"],
+)
+
+scie_binary(
+    name="backendai-plugin-fat",
+    fat=True,
+    dependencies=[":pex"],
+    tags=["scie", "fat"],
+)
+
+resource(name="buildscript", source="BUILD")