Skip to content

Commit

Permalink
Add kleidiai as thirdparty
Browse files Browse the repository at this point in the history
  • Loading branch information
mory91 committed Oct 30, 2024
1 parent 961b891 commit 4b52286
Show file tree
Hide file tree
Showing 4 changed files with 17 additions and 0 deletions.
3 changes: 3 additions & 0 deletions .gitmodules
Original file line number Diff line number Diff line change
Expand Up @@ -90,3 +90,6 @@
[submodule "src/plugins/intel_cpu/thirdparty/shl"]
path = src/plugins/intel_cpu/thirdparty/shl
url = https://github.com/openvinotoolkit/shl.git
[submodule "src/plugins/intel_cpu/thirdparty/kleidiai"]
path = src/plugins/intel_cpu/thirdparty/kleidiai
url = https://git.gitlab.arm.com/kleidi/kleidiai.git
7 changes: 7 additions & 0 deletions src/plugins/intel_cpu/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -271,6 +271,9 @@ endif ()
if(ENABLE_SHL_FOR_CPU)
target_link_libraries(${TARGET_NAME} PRIVATE shl)
endif()
if(ENABLE_KLEIDIAI_FOR_CPU)
target_link_libraries(${TARGET_NAME} PRIVATE kleidiai)
endif()
target_include_directories(${TARGET_NAME} SYSTEM PRIVATE $<TARGET_PROPERTY:dnnl,INCLUDE_DIRECTORIES>)

# Temporal solution to use template reference implementations in cases where optimizied implementation
Expand Down Expand Up @@ -374,6 +377,10 @@ if(BUILD_SHARED_LIBS)
target_include_directories(${TARGET_NAME}_obj SYSTEM PUBLIC $<TARGET_PROPERTY:shl,INTERFACE_INCLUDE_DIRECTORIES>)
endif()

if(ENABLE_KLEIDIAI_FOR_CPU)
target_include_directories(${TARGET_NAME}_obj SYSTEM PUBLIC $<TARGET_PROPERTY:kleidiai,INTERFACE_INCLUDE_DIRECTORIES>)
endif()

ov_set_threading_interface_for(${TARGET_NAME}_obj)

target_compile_definitions(${TARGET_NAME}_obj PRIVATE USE_STATIC_IE)
Expand Down
6 changes: 6 additions & 0 deletions src/plugins/intel_cpu/thirdparty/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -169,4 +169,10 @@ if(ENABLE_SHL_FOR_CPU)
ov_install_static_lib(shl ${OV_CPACK_COMP_CORE})
endif()

if(ENABLE_KLEIDIAI_FOR_CPU)
set(KLEIDIAI_BUILD_TESTS OFF CACHE BOOL "" FORCE)
add_subdirectory(kleidiai)
ov_install_static_lib(kleidiai ${OV_CPACK_COMP_CORE})
endif()

ov_add_onednn()
1 change: 1 addition & 0 deletions src/plugins/intel_cpu/thirdparty/kleidiai
Submodule kleidiai added at 4e83e6

0 comments on commit 4b52286

Please sign in to comment.