46 lines
1.4 KiB
46 lines
1.4 KiB
set(ANAKIN_ROOT "/usr" CACHE PATH "ANAKIN ROOT")
|
|
find_path(ANAKIN_INCLUDE_DIR anakin_config.h
|
|
PATHS ${ANAKIN_ROOT} ${ANAKIN_ROOT}/include
|
|
$ENV{ANAKIN_ROOT} $ENV{ANAKIN_ROOT}/include
|
|
NO_DEFAULT_PATH
|
|
)
|
|
|
|
find_library(ANAKIN_LIBRARY NAMES libanakin_saber_common.so libanakin.so
|
|
PATHS ${ANAKIN_ROOT}
|
|
$ENV{ANAKIN_ROOT} $ENV{ANAKIN_ROOT}/lib
|
|
NO_DEFAULT_PATH
|
|
DOC "Path to ANAKIN library.")
|
|
|
|
if(ANAKIN_INCLUDE_DIR AND ANAKIN_LIBRARY)
|
|
set(ANAKIN_FOUND ON)
|
|
else()
|
|
set(ANAKIN_FOUND OFF)
|
|
endif()
|
|
|
|
if(ANAKIN_FOUND)
|
|
message(STATUS "Current ANAKIN header is ${ANAKIN_INCLUDE_DIR}/anakin_config.h. ")
|
|
include_directories(${ANAKIN_ROOT})
|
|
include_directories(${ANAKIN_ROOT}/include)
|
|
include_directories(${ANAKIN_ROOT}/saber)
|
|
link_directories(${ANAKIN_ROOT})
|
|
add_definitions(-DPADDLE_WITH_ANAKIN)
|
|
endif()
|
|
|
|
if(ANAKIN_FOUND)
|
|
if (ANAKIN_MLU AND NOT WITH_GPU AND NOT ANAKIN_X86)
|
|
message(STATUS "Compile with anakin mlu place.")
|
|
add_definitions(-DANAKIN_MLU_PLACE)
|
|
elseif(ANAKIN_BM AND NOT WITH_GPU AND NOT ANAKIN_X86)
|
|
message(STATUS "Compile with anakin bm place.")
|
|
add_definitions(-DANAKIN_BM_PLACE)
|
|
elseif(ANAKIN_X86)
|
|
message(STATUS "Compile with anakin x86 place.")
|
|
add_definitions(-DANAKIN_X86_PLACE)
|
|
endif()
|
|
endif()
|
|
|
|
if(ANAKIN_FOUND AND WITH_GPU AND WITH_DSO)
|
|
message(STATUS "Compile with anakin subgraph.")
|
|
set(ANAKIN_SUBGRAPH ON)
|
|
endif()
|