zzy34407230
/
mindspore2022
Not watched
1
0
Fork 0
Code
Releases 22 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: 7df661fbaf
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '7df661fbaf'
${ noResults }
mindspore2022/mindspore/ccsrc/cxx_api/CMakeLists.txt

152 lines
7.7 kB
Raw Blame History 

  1. # build mindspore_shared_lib

  2. set(LOAD_MINDIR_SRC

  3.         ${CMAKE_SOURCE_DIR}/mindspore/core/load_mindir/load_model.cc

  4.         ${CMAKE_SOURCE_DIR}/mindspore/core/load_mindir/anf_model_parser.cc

  5.         )

  6. file(GLOB_RECURSE API_OPS_SRC ${CMAKE_CURRENT_SOURCE_DIR} "ops/*.cc")

  7. 

  8. if(ENABLE_ACL)

  9.     add_compile_definitions(ENABLE_ACL)

  10.     include_directories(${CMAKE_SOURCE_DIR}/graphengine/ge)

  11.     include_directories(${CMAKE_BINARY_DIR}/proto/ge)

  12.     file(GLOB_RECURSE API_ACL_SRC ${CMAKE_CURRENT_SOURCE_DIR}

  13.             "akg_kernel_register.cc"

  14.             "model/acl/*.cc"

  15.             "model/model_converter_utils/*.cc"

  16.             "graph/acl/*.cc"

  17.             )

  18. endif()

  19. 

  20. if(ENABLE_D)

  21.     file(GLOB_RECURSE API_MS_INFER_SRC ${CMAKE_CURRENT_SOURCE_DIR}

  22.         "akg_kernel_register.cc" "model/ms/*.cc" "graph/ascend/*.cc")

  23. endif()

  24. 

  25. if(ENABLE_GPU)

  26.     file(GLOB_RECURSE API_MS_INFER_SRC ${CMAKE_CURRENT_SOURCE_DIR}

  27.         "akg_kernel_register.cc" "model/ms/*.cc" "graph/gpu/*.cc")

  28. endif()

  29. 

  30. set(MSLIB_SRC ${CMAKE_CURRENT_SOURCE_DIR}/types.cc

  31.               ${CMAKE_CURRENT_SOURCE_DIR}/context.cc

  32.               ${CMAKE_CURRENT_SOURCE_DIR}/cell.cc

  33.               ${CMAKE_CURRENT_SOURCE_DIR}/serialization.cc

  34.               ${CMAKE_CURRENT_SOURCE_DIR}/graph/graph.cc

  35.               ${CMAKE_CURRENT_SOURCE_DIR}/graph/graph_data.cc

  36.               ${CMAKE_CURRENT_SOURCE_DIR}/model/model.cc

  37.               ${API_MS_INFER_SRC}

  38.               ${API_ACL_SRC}

  39.               ${API_OPS_SRC}

  40.               ${LOAD_MINDIR_SRC})

  41. 

  42. add_library(mindspore_shared_lib SHARED ${MSLIB_SRC})

  43. set_target_properties(mindspore_shared_lib PROPERTIES OUTPUT_NAME mindspore)

  44. 

  45. if(CMAKE_SYSTEM_NAME MATCHES "Darwin")

  46.     target_link_libraries(mindspore_shared_lib PRIVATE ${PYTHON_LIBRARIES} ${SECUREC_LIBRARY}

  47.             -Wl,-force_load mindspore -Wl,-noall_load proto_input mindspore_gvar mindspore::protobuf)

  48. else()

  49.     if(ENABLE_D OR ENABLE_ACL OR ENABLE_GPU)

  50.         target_link_libraries(mindspore_shared_lib PRIVATE ${PYTHON_LIBRARIES} ${SECUREC_LIBRARY}

  51.             -Wl,--whole-archive mindspore -Wl,--no-whole-archive proto_input mindspore_gvar mindspore::protobuf)

  52.     else()

  53.         target_link_libraries(mindspore_shared_lib PRIVATE ${PYTHON_LIBRARIES} ${SECUREC_LIBRARY}

  54.           mindspore proto_input mindspore_gvar mindspore::protobuf)

  55.     endif()

  56. endif()

  57. 

  58. if(ENABLE_CPU)

  59.     target_link_libraries(mindspore_shared_lib PRIVATE mindspore::dnnl mindspore::mkldnn)

  60. endif()

  61. 

  62. if(USE_GLOG)

  63.     target_link_libraries(mindspore_shared_lib PRIVATE mindspore::glog)

  64. endif()

  65. 

  66. if(CMAKE_SYSTEM_NAME MATCHES "Linux")

  67.     target_link_options(mindspore_shared_lib PRIVATE -Wl,-init,common_log_init)

  68. endif()

  69. 

  70. if(ENABLE_ACL)

  71.     if(DEFINED ENV{ASCEND_CUSTOM_PATH})

  72.         set(ASCEND_PATH $ENV{ASCEND_CUSTOM_PATH})

  73.     else()

  74.         set(ASCEND_PATH /usr/local/Ascend)

  75.     endif()

  76.     set(ACL_LIB_DIR ${ASCEND_PATH}/acllib/)

  77.     set(ATLAS_ACL_LIB_DIR ${ASCEND_PATH}/ascend-toolkit/latest/acllib)

  78.     set(ATC_DIR ${ASCEND_PATH}/atc/)

  79.     set(ATLAS_ATC_DIR ${ASCEND_PATH}/ascend-toolkit/latest/atc)

  80.     MESSAGE("acl lib dir " ${ACL_LIB_DIR} ", atc dir " ${ATC_DIR})

  81.     MESSAGE("atlas acl lib dir " ${ATLAS_ACL_LIB_DIR} ", atc dir " ${ATLAS_ATC_DIR})

  82. 

  83.     include_directories(${ACL_LIB_DIR}/include/)

  84.     include_directories(${ATLAS_ACL_LIB_DIR}/include/)

  85.     add_compile_definitions(ENABLE_DVPP_INTERFACE)

  86.     find_library(acl libascendcl.so ${ACL_LIB_DIR}/lib64 ${ATLAS_ACL_LIB_DIR}/lib64)

  87.     find_library(acl_retr libacl_retr.so ${ACL_LIB_DIR}/lib64 ${ATLAS_ACL_LIB_DIR}/lib64)

  88.     find_library(acl_cblas libacl_cblas.so ${ACL_LIB_DIR}/lib64 ${ATLAS_ACL_LIB_DIR}/lib64)

  89.     find_library(acl_dvpp libacl_dvpp.so ${ACL_LIB_DIR}/lib64 ${ATLAS_ACL_LIB_DIR}/lib64)

  90.     find_library(acl_runtime libruntime.so ${ACL_LIB_DIR}/lib64 ${ATLAS_ACL_LIB_DIR}/lib64)

  91.     find_library(ge_compiler libge_compiler.so ${ATC_DIR}/lib64 ${ATLAS_ATC_DIR}/lib64)

  92.     find_library(libplatform libplatform.so ${ATC_DIR}/lib64 ${ATLAS_ATC_DIR}/lib64)

  93.     find_library(libcompress libcompress.so ${ATC_DIR}/lib64 ${ATLAS_ATC_DIR}/lib64)

  94.     find_library(libopskernel libopskernel.so ${ATC_DIR}/lib64 ${ATLAS_ATC_DIR}/lib64)

  95.     find_library(libaicore_utils libaicore_utils.so ${ATC_DIR}/lib64 ${ATLAS_ATC_DIR}/lib64)

  96.     find_library(libaicpu_engine_common libaicpu_engine_common.so ${ATC_DIR}/lib64 ${ATLAS_ATC_DIR}/lib64)

  97.     find_library(atc_protobuf libascend_protobuf.so.3.8.0.0 ${ATC_DIR}/lib64 ${ATLAS_ATC_DIR}/lib64)

  98. 

  99.     target_link_libraries(mindspore_shared_lib PRIVATE -Wl,--no-as-needed graph ${ge_compiler} ${atc_protobuf}

  100.             ${acl_retr} ${acl_cblas} ${acl_dvpp} ${acl_runtime} ${libplatform} ${libcompress} ${libopskernel}

  101.             ${libaicore_utils} ${libaicpu_engine_common} ${acl})

  102.     add_dependencies(mindspore_shared_lib graph)

  103. endif()

  104. 

  105. 

  106. # Before build inference

  107. if(ENABLE_D)

  108.     find_library(adump_server libadump_server.a ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH})

  109.     target_link_libraries(mindspore_shared_lib PRIVATE ${adump_server})

  110. endif()

  111. 

  112. if(ENABLE_GPU)

  113.     target_link_libraries(mindspore_shared_lib PRIVATE  gpu_cuda_lib gpu_queue cublas

  114.                           ${CUDA_PATH}/lib64/libcurand.so

  115.                           ${CUDNN_LIBRARY_PATH}

  116.                           ${CUDA_PATH}/lib64/libcudart.so

  117.                           ${CUDA_PATH}/lib64/stubs/libcuda.so

  118.                           ${CUDA_PATH}/lib64/libcusolver.so

  119.                           ${CUDA_PATH}/lib64/libcufft.so)

  120. endif()

  121. 

  122. if(CMAKE_SYSTEM_NAME MATCHES "Linux")

  123.     set(MINDSPORE_RPATH $ORIGIN)

  124.     if(ENABLE_D)

  125.         set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/nnae/latest/fwkacllib/lib64)

  126.         set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/fwkacllib/lib64)

  127.         set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/fwkacllib/lib64)

  128.         set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/nnae/latest/fwkacllib/lib64/plugin/opskernel)

  129.         set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/fwkacllib/lib64/plugin/opskernel)

  130.         set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/fwkacllib/lib64/plugin/opskernel)

  131.         set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/add-ons)

  132.         set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe/op_tiling)

  133.         set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/nnae/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling)

  134.         set(MINDSPORE_RPATH

  135.             ${MINDSPORE_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling)

  136.     elseif(ENABLE_ACL)

  137.         set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/nnae/latest/atc/lib64)

  138.         set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/atc/lib64)

  139.         set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/atc/lib64)

  140.         set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/nnae/latest/acllib/lib64)

  141.         set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/acllib/lib64)

  142.         set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/acllib/lib64)

  143.         set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/add-ons)

  144.         set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe/op_tiling)

  145.         set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/nnae/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling)

  146.         set(MINDSPORE_RPATH

  147.             ${MINDSPORE_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling)

  148.     endif()

  149. 

  150.     set_target_properties(mindspore_shared_lib PROPERTIES INSTALL_RPATH ${MINDSPORE_RPATH})

  151. endif()