up

Author: metascroy

torchchat/utils/scripts/install_utils.sh

@@ -200,6 +200,7 @@ install_torchao_aten_ops() {
   CMAKE_OUT_DIR=${TORCHCHAT_ROOT}/torchao-build/cmake-out
   cmake -DCMAKE_PREFIX_PATH=${MY_CMAKE_PREFIX_PATH} \
     -DCMAKE_INSTALL_PREFIX=${CMAKE_OUT_DIR} \
+    -DTORCHAO_BUILD_CPU_AARCH64=ON \
     -DCMAKE_BUILD_TYPE="Release" \
     -S . \
     -B ${CMAKE_OUT_DIR} -G Ninja
@@ -217,6 +218,7 @@ install_torchao_executorch_ops() {
     -DCMAKE_INSTALL_PREFIX=${CMAKE_OUT_DIR} \
     -DCMAKE_BUILD_TYPE="Release" \
     -DTORCHAO_BUILD_EXECUTORCH_OPS=ON \
+    -DTORCHAO_BUILD_CPU_AARCH64=ON \
     -DEXECUTORCH_INCLUDE_DIRS="${EXECUTORCH_INCLUDE_DIRS}" \
     -DEXECUTORCH_LIBRARIES="${EXECUTORCH_LIBRARIES}" \
     -S . \