parent
2d529186f1
commit
3ce10dba15
@ -1,30 +0,0 @@
|
||||
# Find the NNPACK library
|
||||
# NNPACK_ROOT - where to find NNPACK include and library.
|
||||
#
|
||||
|
||||
set(NNPACK_FOUND OFF)
|
||||
set(NNPACK_ROOT $ENV{NNPACK_ROOT} CACHE PATH "Folder contains NNPACK")
|
||||
find_path(NNPACK_INC_DIR nnpack.h PATHS ${NNPACK_ROOT}/include)
|
||||
find_library(NNPACK_LIB NAMES nnpack PATHS ${NNPACK_ROOT}/lib)
|
||||
find_library(PTHREADPOOL_LIB NAMES pthreadpool PATHS ${NNPACK_ROOT}/lib)
|
||||
find_library(NNPACK_UKERNELS_LIB NAMES nnpack_ukernels PATHS ${NNPACK_ROOT}/lib)
|
||||
find_library(NNPACK_CPUFEATURES_LIB NAMES cpufeatures PATHS ${NNPACK_ROOT}/lib)
|
||||
|
||||
if(NNPACK_INC_DIR AND NNPACK_LIB AND PTHREADPOOL_LIB)
|
||||
set(NNPACK_FOUND ON)
|
||||
INCLUDE_DIRECTORIES(${NNPACK_INC_DIR})
|
||||
|
||||
set(NNPACK_LIBS)
|
||||
list(APPEND NNPACK_LIBS ${NNPACK_LIB} ${PTHREADPOOL_LIB})
|
||||
if (NNPACK_UKERNELS_LIB)
|
||||
list(APPEND NNPACK_LIBS ${NNPACK_UKERNELS_LIB})
|
||||
endif()
|
||||
if (NNPACK_CPUFEATURES_LIB)
|
||||
list(APPEND NNPACK_LIBS ${NNPACK_CPUFEATURES_LIB})
|
||||
endif()
|
||||
if(NOT ANDROID)
|
||||
list(APPEND NNPACK_LIBS "rt")
|
||||
endif()
|
||||
else()
|
||||
message(FATAL_ERROR "Cannot find NNPACK in (${NNPACK_ROOT})")
|
||||
endif()
|
Loading…
Reference in new issue