updated cmake and framework build

TODO move some of these to the apple build framework

Author: hinriksnaer

CMakeLists.txt

@@ -246,13 +246,24 @@ if(EXECUTORCH_BUILD_KERNELS_TORCHAO)
   set(TORCHAO_BUILD_EXECUTORCH_OPS ON)
   set(TORCHAO_BUILD_CPU_AARCH64 ON)
   set(TORCHAO_ENABLE_ARM_NEON_DOT ON)
+  set(TORCHAO_BUILD_ATEN_OPS OFF)
 
   # Set ExecuTorch headers and core library so TorchAO can register ops against them
   set(EXECUTORCH_INCLUDE_DIRS ${CMAKE_CURRENT_SOURCE_DIR}/..)
   set(EXECUTORCH_LIBRARIES executorch)
+
   add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/third-party/ao/torchao/experimental)
+
+  # Ensure torchao_ops_executorch can find and link pthreadpool
+  target_include_directories(torchao_ops_executorch PRIVATE ${PTHREADPOOL_SOURCE_DIR}/include)
+  target_link_libraries(torchao_ops_executorch PRIVATE pthreadpool)
+
+  # Only call this on torchao_ops_executorch — pthreadpool is not a shared lib
   target_link_options_shared_lib(torchao_ops_executorch)
+
+  # Ensure pthreadpool is included in the packaged framework
   list(APPEND _dep_libs torchao_ops_executorch)
+  list(APPEND _dep_libs pthreadpool)
 endif()
 
 if(EXECUTORCH_BUILD_TESTS)

scripts/build_apple_frameworks.sh

@@ -163,6 +163,7 @@ cmake_build() {
     -DCMAKE_BUILD_TYPE="$mode" \
     -DCMAKE_XCODE_ATTRIBUTE_CODE_SIGNING_ALLOWED=NO \
     -DCMAKE_XCODE_ATTRIBUTE_CODE_SIGNING_REQUIRED=NO \
+    -DEXECUTORCH_BUILD_PTHREADPOOL=ON \
     -DCMAKE_XCODE_ATTRIBUTE_CODE_SIGN_IDENTITY="" \
     -DCMAKE_TOOLCHAIN_FILE="$TOOLCHAIN" \
     -DCMAKE_XCODE_ATTRIBUTE_CLANG_CXX_LANGUAGE_STANDARD="c++17" \