forked from huawei/mindspore2022
180 lines
8.3 KiB
CMake
180 lines
8.3 KiB
CMake
# find python library
|
|
if(MODE_ASCEND_ACL)
|
|
get_filename_component(PYTHON_LIB_REALPATH ${PYTHON_LIBRARIES} ABSOLUTE)
|
|
get_filename_component(PYTHON_LIB_DIR ${PYTHON_LIB_REALPATH} PATH)
|
|
|
|
execute_process(
|
|
COMMAND "${Python3_EXECUTABLE}" -c "import distutils.sysconfig as sysconfig; \
|
|
print(sysconfig.get_config_var('PY3LIBRARY'))"
|
|
RESULT_VARIABLE result
|
|
OUTPUT_VARIABLE PY3LIBRARY)
|
|
set(PY3_LIBG ${PYTHON_LIB_DIR}/${PY3LIBRARY})
|
|
string(STRIP "${PY3_LIBG}" PY3_LIBG)
|
|
message("Python3 general library = " ${PY3_LIBG})
|
|
endif()
|
|
# build mindspore_shared_lib
|
|
include_directories(${CMAKE_SOURCE_DIR}/mindspore/ccsrc)
|
|
include_directories(${CMAKE_SOURCE_DIR}/mindspore/ccsrc/minddata/dataset)
|
|
|
|
file(GLOB_RECURSE API_OPS_SRC ${CMAKE_CURRENT_SOURCE_DIR} "ops/*.cc")
|
|
|
|
if(ENABLE_D OR ENABLE_ACL)
|
|
# build 910 and 310 code into one distro, files needed for 310 mode
|
|
add_compile_definitions(ENABLE_ACL)
|
|
include_directories(${CMAKE_SOURCE_DIR}/graphengine/ge)
|
|
include_directories(${CMAKE_BINARY_DIR}/proto/ge)
|
|
|
|
file(GLOB_RECURSE API_ACL_SRC ${CMAKE_CURRENT_SOURCE_DIR}
|
|
"akg_kernel_register.cc"
|
|
"model/acl/*.cc"
|
|
"model/model_converter_utils/*.cc"
|
|
"graph/acl/*.cc"
|
|
)
|
|
endif()
|
|
|
|
if(ENABLE_D)
|
|
# for 910 mode
|
|
file(GLOB_RECURSE API_MS_INFER_SRC ${CMAKE_CURRENT_SOURCE_DIR}
|
|
"model/ms/*.cc" "graph/ascend/*.cc")
|
|
endif()
|
|
|
|
if(ENABLE_GPU)
|
|
file(GLOB_RECURSE API_MS_INFER_SRC ${CMAKE_CURRENT_SOURCE_DIR}
|
|
"akg_kernel_register.cc" "model/ms/*.cc" "graph/gpu/*.cc")
|
|
endif()
|
|
|
|
set(MSLIB_SRC ${CMAKE_CURRENT_SOURCE_DIR}/types.cc
|
|
${CMAKE_CURRENT_SOURCE_DIR}/context.cc
|
|
${CMAKE_CURRENT_SOURCE_DIR}/cell.cc
|
|
${CMAKE_CURRENT_SOURCE_DIR}/serialization.cc
|
|
${CMAKE_CURRENT_SOURCE_DIR}/graph/graph.cc
|
|
${CMAKE_CURRENT_SOURCE_DIR}/graph/graph_data.cc
|
|
${CMAKE_CURRENT_SOURCE_DIR}/model/model.cc
|
|
${CMAKE_CURRENT_SOURCE_DIR}/model/model_impl.cc
|
|
${API_MS_INFER_SRC}
|
|
${API_ACL_SRC}
|
|
${API_OPS_SRC}
|
|
)
|
|
|
|
if(NOT ENABLE_TESTCASES AND NOT BUILD_LITE)
|
|
# users of shared_lib cannot find symbols in indirect dependency
|
|
set(MSLIB_SRC ${MSLIB_SRC} ${CMAKE_SOURCE_DIR}/mindspore/core/utils/status.cc)
|
|
endif()
|
|
|
|
if(BUILD_LITE)
|
|
list(APPEND MSLIB_SRC "${CMAKE_CURRENT_SOURCE_DIR}/../../../mindspore/ccsrc/utils/config_manager.cc")
|
|
file(GLOB_RECURSE ACL_REMOVE_SRC ${CMAKE_CURRENT_SOURCE_DIR}
|
|
"model/acl/acl_vm/*.cc"
|
|
)
|
|
list(REMOVE_ITEM MSLIB_SRC "${CMAKE_CURRENT_SOURCE_DIR}/akg_kernel_register.cc"
|
|
"${CMAKE_CURRENT_SOURCE_DIR}/model/acl/acl_model_multi.cc"
|
|
"${CMAKE_CURRENT_SOURCE_DIR}/model/acl/acl_model.cc"
|
|
"${CMAKE_CURRENT_SOURCE_DIR}/serialization.cc"
|
|
"${CMAKE_CURRENT_SOURCE_DIR}/types.cc"
|
|
"${CMAKE_CURRENT_SOURCE_DIR}/model/model.cc"
|
|
"${CMAKE_CURRENT_SOURCE_DIR}/model/model_impl.cc"
|
|
${ACL_REMOVE_SRC})
|
|
endif()
|
|
|
|
add_library(mindspore_shared_lib_obj OBJECT ${MSLIB_SRC})
|
|
target_link_libraries(mindspore_shared_lib_obj PRIVATE mindspore::protobuf mindspore::grpc++ mindspore::json
|
|
mindspore::pybind11_module)
|
|
|
|
add_library(mindspore_shared_lib SHARED $<TARGET_OBJECTS:mindspore_shared_lib_obj>)
|
|
target_link_libraries(mindspore_shared_lib_obj PUBLIC mindspore::opencv_core $<$<BOOL:${ENABLE_GPU}>:CUDA::cuda_driver>)
|
|
if(BUILD_LITE)
|
|
target_link_libraries(mindspore_shared_lib PRIVATE $<TARGET_OBJECTS:_mindspore_transform_graph_ir_obj>)
|
|
add_dependencies(mindspore_shared_lib _mindspore_transform_graph_ir_obj)
|
|
elseif(ENABLE_ACL AND NOT ENABLE_D)
|
|
target_link_libraries(mindspore_shared_lib PRIVATE $<TARGET_OBJECTS:_mindspore_backend_graph_compiler_obj>)
|
|
endif()
|
|
if(NOT BUILD_LITE)
|
|
set_target_properties(mindspore_shared_lib PROPERTIES OUTPUT_NAME mindspore)
|
|
endif()
|
|
|
|
if(ENABLE_D OR ENABLE_GPU)
|
|
target_link_libraries(mindspore_shared_lib PRIVATE -Wl,--as-needed ${PYTHON_LIBRARIES} ${SECUREC_LIBRARY}
|
|
-Wl,--whole-archive mindspore -Wl,--no-whole-archive mindspore_core mindspore_common proto_input
|
|
mindspore::protobuf)
|
|
else()
|
|
if(BUILD_LITE)
|
|
target_link_libraries(mindspore_shared_lib PRIVATE mindspore_core ${SECUREC_LIBRARY})
|
|
else()
|
|
target_link_libraries(mindspore_shared_lib PRIVATE ${PY3_LIBG} ${SECUREC_LIBRARY}
|
|
mindspore mindspore_core mindspore_common proto_input mindspore::protobuf)
|
|
endif()
|
|
endif()
|
|
|
|
if(ENABLE_CPU)
|
|
target_link_libraries(mindspore_shared_lib PRIVATE mindspore::dnnl mindspore::mkldnn nnacl)
|
|
endif()
|
|
|
|
if(USE_GLOG)
|
|
target_link_libraries(mindspore_shared_lib_obj PRIVATE mindspore::glog)
|
|
target_link_libraries(mindspore_shared_lib PRIVATE mindspore::glog)
|
|
endif()
|
|
|
|
if(CMAKE_SYSTEM_NAME MATCHES "Linux")
|
|
target_link_options(mindspore_shared_lib PRIVATE -Wl,-init,common_log_init)
|
|
endif()
|
|
|
|
if(ENABLE_D OR ENABLE_ACL)
|
|
# 310 mode
|
|
add_compile_definitions(ENABLE_DVPP_INTERFACE)
|
|
find_library(acl libascendcl.so ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH})
|
|
find_library(acl_cblas libacl_cblas.so ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH})
|
|
find_library(acl_dvpp libacl_dvpp.so ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH})
|
|
find_library(acl_runtime libruntime.so ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH})
|
|
find_library(ge_compiler libge_compiler.so ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH})
|
|
find_library(libplatform libplatform.so ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH})
|
|
find_library(libcompress libcompress.so ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH})
|
|
find_library(libopskernel libopskernel.so ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH})
|
|
find_library(libaicore_utils libaicore_utils.so ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH})
|
|
find_library(libaicpu_engine_common libaicpu_engine_common.so ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH})
|
|
|
|
target_link_libraries(mindspore_shared_lib PRIVATE -Wl,--no-as-needed graph ${ge_compiler}
|
|
${acl_retr} ${acl_cblas} ${acl_dvpp} ${acl_runtime} ${libplatform} ${libcompress} ${libopskernel}
|
|
${libaicore_utils} ${libaicpu_engine_common} ${acl})
|
|
add_dependencies(mindspore_shared_lib_obj graph)
|
|
add_dependencies(mindspore_shared_lib graph)
|
|
endif()
|
|
|
|
|
|
# Before build inference
|
|
if(ENABLE_D)
|
|
find_library(adump_server libadump_server.a ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH})
|
|
target_link_libraries(mindspore_shared_lib PRIVATE ${adump_server})
|
|
endif()
|
|
|
|
if(ENABLE_GPU)
|
|
target_link_libraries(mindspore_shared_lib PRIVATE gpu_cuda_lib gpu_queue CUDA::cublas cuda_ops
|
|
CUDA::curand
|
|
${CUDNN_LIBRARY_PATH}
|
|
CUDA::cudart
|
|
CUDA::cuda_driver
|
|
CUDA::cusolver
|
|
CUDA::cufft)
|
|
endif()
|
|
|
|
if(CMAKE_SYSTEM_NAME MATCHES "Linux")
|
|
set(MINDSPORE_RPATH $ORIGIN)
|
|
if(ENABLE_D OR ENABLE_ACL)
|
|
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/nnae/latest/fwkacllib/lib64)
|
|
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/fwkacllib/lib64)
|
|
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/fwkacllib/lib64)
|
|
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe/op_tiling)
|
|
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/nnae/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling)
|
|
set(MINDSPORE_RPATH
|
|
${MINDSPORE_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling)
|
|
|
|
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/nnae/latest/atc/lib64)
|
|
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/atc/lib64)
|
|
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/atc/lib64)
|
|
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/nnae/latest/acllib/lib64)
|
|
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/acllib/lib64)
|
|
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/acllib/lib64)
|
|
endif()
|
|
|
|
set_target_properties(mindspore_shared_lib PROPERTIES INSTALL_RPATH ${MINDSPORE_RPATH})
|
|
endif()
|