mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2024-11-24 03:00:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

link libacl_dvpp_mpi library when building with CANN backend

Browse Source
This commit is contained in:
MengqingCao 2023-12-19 17:35:11 +08:00
parent 408730b7ab
commit b6cc74f855
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
cmake/OpenCVFindCANN.cmake
View File

@ -57,6 +57,18 @@ if(CANN_INSTALL_DIR)
set(HAVE_CANN OFF) set(HAVE_CANN OFF)
return() return()
endif() endif()
# * libacl_dvpp_mpi.so
set(libacl_dvpp_mpi "${CANN_INSTALL_DIR}/lib64")
find_library(found_libacldvppmpi NAMES acl_dvpp_mpi PATHS ${libacl_dvpp_mpi} NO_DEFAULT_PATH)
if(found_libacldvppmpi)
set(libacl_dvpp_mpi ${found_libacldvppmpi})
message(STATUS "CANN: libacl_dvpp_mpi.so is found at ${libacl_dvpp_mpi}")
else()
message(STATUS "CANN: Missing libacl_dvpp_mpi.so. Turning off HAVE_CANN")
set(HAVE_CANN OFF)
return()
endif()
# * libgraph.so # * libgraph.so
set(lib_graph "${CANN_INSTALL_DIR}/compiler/lib64") set(lib_graph "${CANN_INSTALL_DIR}/compiler/lib64")
find_library(found_lib_graph NAMES graph PATHS ${lib_graph} NO_DEFAULT_PATH) find_library(found_lib_graph NAMES graph PATHS ${lib_graph} NO_DEFAULT_PATH)
@ -105,6 +117,7 @@ if(CANN_INSTALL_DIR)
list(APPEND libs_cann ${lib_opsproto}) list(APPEND libs_cann ${lib_opsproto})
list(APPEND libs_cann ${lib_graph}) list(APPEND libs_cann ${lib_graph})
list(APPEND libs_cann ${lib_ge_compiler}) list(APPEND libs_cann ${lib_ge_compiler})
list(APPEND libs_cann ${libacl_dvpp_mpi})
# * lib_graph_base.so # * lib_graph_base.so
if(NOT CANN_VERSION_BELOW_6_3_ALPHA002) if(NOT CANN_VERSION_BELOW_6_3_ALPHA002)