fix: add target option

elixir-nx · Jul 16, 2024 · fee1738 · fee1738
1 parent 7ef3c61
commit fee1738
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 5 deletions.
diff --git a/.github/workflows/embedded_devices.yml b/.github/workflows/embedded_devices.yml
@@ -64,7 +64,7 @@ jobs:
         if: steps.cache-iree-dir.outputs.cache-hit != 'true'
         run: make clone_iree
       - name: Compile
-        run: ./scripts/build_and_package.sh ${{ matrix.build_target }} --build-host
+        run: ./scripts/build_and_package.sh --target=${{ matrix.build_target }} --build-host
       - name: Release Artifacts
         uses: softprops/action-gh-release@v2
         if: startsWith(github.ref, 'refs/tags/')

diff --git a/scripts/build_and_package.sh b/scripts/build_and_package.sh
@@ -2,8 +2,6 @@
 
 set -e
 
-IREE_BUILD_TARGET=$1
-
 get_nproc() {
     if [ "$(uname -s)" = "Darwin" ]; then
         sysctl -n hw.ncpu
@@ -18,6 +16,10 @@ build_host_flag=false
 # Parse options
 while [[ $# -gt 0 ]]; do
   case $1 in
+    --target=*)
+      IREE_BUILD_TARGET="${1#*=}"
+      shift # Shift past the argument
+      ;;
     --build-host)
       build_host_flag=true
       shift # Shift past the argument

diff --git a/scripts/build_linux.sh b/scripts/build_linux.sh
@@ -5,6 +5,6 @@ do
 
 SCRIPT_DIR=$(dirname "$0")
 
-${SCRIPT_DIR}/build_and_package.sh ${IREE_BUILD_TARGET}
+${SCRIPT_DIR}/build_and_package.sh --target=${IREE_BUILD_TARGET}
 
 done
diff --git a/scripts/build_mac_arm.sh b/scripts/build_mac_arm.sh
@@ -5,6 +5,6 @@ do
 
 SCRIPT_DIR=$(dirname "$0")
 
-${SCRIPT_DIR}/build_and_package.sh ${IREE_BUILD_TARGET}
+${SCRIPT_DIR}/build_and_package.sh --target=${IREE_BUILD_TARGET}
 
 done