up

Author: metascroy

CMakeLists.txt

@@ -243,27 +243,27 @@ endif()
 
 if(EXECUTORCH_BUILD_KERNELS_TORCHAO)
   # Currently only enable this on Arm-based Macs
+  set(TORCHAO_BUILD_ATEN_OPS OFF)
   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)
+  set(EXECUTORCH_LIBRARIES executorch pthreadpool extension_threadpool cpuinfo)
 
   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)
+  # # 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)
+  # list(APPEND _dep_libs pthreadpool)
 endif()
 
 if(EXECUTORCH_BUILD_TESTS)

scripts/build_apple_frameworks.sh

@@ -159,7 +159,6 @@ cmake_build() {
   echo "Building for $platform ($mode) with flag $platform_flag"
   mkdir -p "$platform" && cd "$platform" || exit 1
   cmake "$SOURCE_ROOT_DIR" -G Xcode \
-    -DCMAKE_PREFIX_PATH=$(python -c 'from distutils.sysconfig import get_python_lib; print(get_python_lib())') \
     -DCMAKE_BUILD_TYPE="$mode" \
     -DCMAKE_XCODE_ATTRIBUTE_CODE_SIGNING_ALLOWED=NO \
     -DCMAKE_XCODE_ATTRIBUTE_CODE_SIGNING_REQUIRED=NO \