File tree Expand file tree Collapse file tree 1 file changed +9
-3
lines changed
Expand file tree Collapse file tree 1 file changed +9
-3
lines changed Original file line number Diff line number Diff 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)
292298endif ()
293299
294300if (EXECUTORCH_BUILD_TESTS)
You can’t perform that action at this time.
0 commit comments