blob: c727905a64954e2d2a8da16e8ae796517f3044fc (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
nnfw_find_package(ARMCompute QUIET)
if(NOT ARMCompute_FOUND)
return()
endif(NOT ARMCompute_FOUND)
function(add_kben_acl_cl_library)
cmake_parse_arguments(ARG "" "NAME" "SOURCES" ${ARGN})
add_library(${ARG_NAME} SHARED ${ARG_SOURCES})
target_compile_options(${ARG_NAME} PRIVATE -Wno-psabi)
target_include_directories(${ARG_NAME} PUBLIC ${CMAKE_CURRENT_SOURCE_DIR}/..)
target_link_libraries(${ARG_NAME} nonius)
target_link_libraries(${ARG_NAME} arm_compute)
target_link_libraries(${ARG_NAME} arm_compute_ex)
target_link_libraries(${ARG_NAME} pthread)
install(TARGETS ${ARG_NAME} DESTINATION lib/kben)
endfunction(add_kben_acl_cl_library)
add_kben_acl_cl_library(NAME kben_acl_cl_conv SOURCES Convolution.cpp)
add_kben_acl_cl_library(NAME kben_acl_cl_transpose_conv SOURCES TransposeConv.cpp)
|