Skip to content

Commit eab0f73

Browse files
committed
changed dependency reference in torchao build
1 parent 03f584a commit eab0f73

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

CMakeLists.txt

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -283,12 +283,18 @@ if(EXECUTORCH_BUILD_KERNELS_TORCHAO)
283283
set(TORCHAO_BUILD_EXECUTORCH_OPS ON)
284284
set(TORCHAO_BUILD_CPU_AARCH64 ON)
285285
set(TORCHAO_ENABLE_ARM_NEON_DOT ON)
286-
set(EXECUTORCH_INCLUDE_DIRS ${CMAKE_CURRENT_SOURCE_DIR}/..)
287-
set(EXECUTORCH_LIBRARIES executorch pthreadpool extension_threadpool cpuinfo)
286+
287+
list(APPEND TORCHAO_INCLUDE_DIRS
288+
${EXECUTORCH_ROOT}/backends/xnnpack/third-party/pthreadpool/include
289+
${EXECUTORCH_ROOT}/backends/xnnpack/third-party/cpuinfo/include
290+
${EXECUTORCH_ROOT}/third-party/ao
291+
)
292+
293+
set(EXECUTORCH_INCLUDE_DIRS ${TORCHAO_INCLUDE_DIRS})
288294

289295
add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/third-party/ao/torchao/experimental)
290296
executorch_target_link_options_shared_lib(torchao_ops_executorch)
291-
list(APPEND _dep_libs torchao_ops_executorch)
297+
list(APPEND _executorch_kernels torchao_ops_executorch)
292298
endif()
293299

294300
if(EXECUTORCH_BUILD_TESTS)

0 commit comments

Comments
 (0)