test

Author: jathu

backends/apple/coreml/scripts/install_requirements.sh

@@ -5,58 +5,58 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-SCRIPT_DIR_PATH="$(
-    cd -- "$(dirname "$0")" >/dev/null 2>&1
-    pwd -P
-)"
-
-# TODO(jathu): remove the need to fetch coremltools to build deps for coreml_executor_runner.
-# Keep this version in sync with: pyproject.toml
-COREMLTOOLS_VERSION="8.1"
-
-red=`tput setaf 1`
-green=`tput setaf 2`
-
-EXECUTORCH_ROOT_PATH=$(realpath "$SCRIPT_DIR_PATH/../../../../")
-COREML_DIR_PATH="$EXECUTORCH_ROOT_PATH/backends/apple/coreml"
-COREMLTOOLS_DIR_PATH="$COREML_DIR_PATH/third-party/coremltools"
-PROTOBUF_FILES_DIR_PATH="$COREMLTOOLS_DIR_PATH/build/mlmodel/format/"
-
-cd "$EXECUTORCH_ROOT_PATH"
-
-rm -rf "$COREML_DIR_PATH/third-party"
-mkdir "$COREML_DIR_PATH/third-party"
-
-echo "${green}ExecuTorch: Cloning coremltools."
-git clone --depth 1 --branch "${COREMLTOOLS_VERSION}" "https://github.com/apple/coremltools.git" $COREMLTOOLS_DIR_PATH
-cd $COREMLTOOLS_DIR_PATH
-
-STATUS=$?
-if [ $STATUS -ne 0 ]; then
-    echo "${red}ExecuTorch: Failed to clone coremltools."
-    exit 1
-fi
-
-echo "${green}ExecuTorch: Installing coremltools dependencies."
-pip install -r "$COREMLTOOLS_DIR_PATH/reqs/build.pip"
-STATUS=$?
-if [ $STATUS -ne 0 ]; then
-    echo "${red}ExecuTorch: Failed to install coremltools dependencies."
-    exit 1
-fi
-
-mkdir "$COREMLTOOLS_DIR_PATH/build"
-cmake -S "$COREMLTOOLS_DIR_PATH" -B "$COREMLTOOLS_DIR_PATH/build"
-cmake --build "$COREMLTOOLS_DIR_PATH/build" --parallel --target mlmodel
-
-echo "${green}ExecuTorch: Cloning nlohmann."
-git clone https://github.com/nlohmann/json.git "$COREML_DIR_PATH/third-party/nlohmann_json"
-STATUS=$?
-if [ $STATUS -ne 0 ]; then
-    echo "${red}ExecuTorch: Failed to clone nlohmann."
-    exit 1
-fi
-
-echo "${green}ExecuTorch: Copying protobuf files."
-mkdir -p "$COREML_DIR_PATH/runtime/sdk/format/"
-cp -rf "$PROTOBUF_FILES_DIR_PATH" "$COREML_DIR_PATH/runtime/sdk/format/"
+# SCRIPT_DIR_PATH="$(
+#     cd -- "$(dirname "$0")" >/dev/null 2>&1
+#     pwd -P
+# )"
+
+# # TODO(jathu): remove the need to fetch coremltools to build deps for coreml_executor_runner.
+# # Keep this version in sync with: pyproject.toml
+# COREMLTOOLS_VERSION="8.1"
+
+# red=`tput setaf 1`
+# green=`tput setaf 2`
+
+# EXECUTORCH_ROOT_PATH=$(realpath "$SCRIPT_DIR_PATH/../../../../")
+# COREML_DIR_PATH="$EXECUTORCH_ROOT_PATH/backends/apple/coreml"
+# COREMLTOOLS_DIR_PATH="$COREML_DIR_PATH/third-party/coremltools"
+# PROTOBUF_FILES_DIR_PATH="$COREMLTOOLS_DIR_PATH/build/mlmodel/format/"
+
+# cd "$EXECUTORCH_ROOT_PATH"
+
+# rm -rf "$COREML_DIR_PATH/third-party"
+# mkdir "$COREML_DIR_PATH/third-party"
+
+# echo "${green}ExecuTorch: Cloning coremltools."
+# git clone --depth 1 --branch "${COREMLTOOLS_VERSION}" "https://github.com/apple/coremltools.git" $COREMLTOOLS_DIR_PATH
+# cd $COREMLTOOLS_DIR_PATH
+
+# STATUS=$?
+# if [ $STATUS -ne 0 ]; then
+#     echo "${red}ExecuTorch: Failed to clone coremltools."
+#     exit 1
+# fi
+
+# echo "${green}ExecuTorch: Installing coremltools dependencies."
+# pip install -r "$COREMLTOOLS_DIR_PATH/reqs/build.pip"
+# STATUS=$?
+# if [ $STATUS -ne 0 ]; then
+#     echo "${red}ExecuTorch: Failed to install coremltools dependencies."
+#     exit 1
+# fi
+
+# mkdir "$COREMLTOOLS_DIR_PATH/build"
+# cmake -S "$COREMLTOOLS_DIR_PATH" -B "$COREMLTOOLS_DIR_PATH/build"
+# cmake --build "$COREMLTOOLS_DIR_PATH/build" --parallel --target mlmodel
+
+# echo "${green}ExecuTorch: Cloning nlohmann."
+# git clone https://github.com/nlohmann/json.git "$COREML_DIR_PATH/third-party/nlohmann_json"
+# STATUS=$?
+# if [ $STATUS -ne 0 ]; then
+#     echo "${red}ExecuTorch: Failed to clone nlohmann."
+#     exit 1
+# fi
+
+# echo "${green}ExecuTorch: Copying protobuf files."
+# mkdir -p "$COREML_DIR_PATH/runtime/sdk/format/"
+# cp -rf "$PROTOBUF_FILES_DIR_PATH" "$COREML_DIR_PATH/runtime/sdk/format/"

examples/apple/coreml/scripts/build_executor_runner.sh

@@ -42,8 +42,9 @@ cmake "$EXECUTORCH_ROOT_PATH" -B"$CMAKE_BUILD_DIR_PATH" \
 -DCOREML_BUILD_EXECUTOR_RUNNER=ON \
 -DCMAKE_MACOSX_BUNDLE=OFF
 
-cmake --build "$CMAKE_BUILD_DIR_PATH" -j9 -t coremldelegate
-cmake --build "$CMAKE_BUILD_DIR_PATH" -j9 -t etdump -t flatccrt
+core_count=$(sysctl -n hw.ncpu)
+cmake --build "$CMAKE_BUILD_DIR_PATH" -j$(core_count) -t coremldelegate
+cmake --build "$CMAKE_BUILD_DIR_PATH" -j$(core_count) -t etdump -t flatccrt
 
 # Copy CoreML delegate headers
 echo "ExecuTorch: Copying headers"