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.

protobuf_shared.cmake 3.4 kB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  1. if (HAVE_PROTOBUF)
  2. return()
  3. endif()
  4. include(ExternalProject)
  5. include(GNUInstallDirs)
  6. if ((${CMAKE_INSTALL_PREFIX} STREQUAL /usr/local) OR
  7. (${CMAKE_INSTALL_PREFIX} STREQUAL "C:/Program Files (x86)/ascend"))
  8. set(CMAKE_INSTALL_PREFIX ${GE_CODE_DIR}/output CACHE STRING "path for install()" FORCE)
  9. message(STATUS "No install prefix selected, default to ${CMAKE_INSTALL_PREFIX}.")
  10. endif()
  11. if (ENABLE_GITEE)
  12. set(REQ_URL "https://gitee.com/mirrors/protobuf_source/repository/archive/v3.8.0.tar.gz")
  13. set(MD5 "eba86ae9f07ba5cfbaf8af3bc4e84236")
  14. else()
  15. set(REQ_URL "https://github.com/protocolbuffers/protobuf/archive/v3.8.0.tar.gz")
  16. set(MD5 "3d9e32700639618a4d2d342c99d4507a")
  17. endif ()
  18. set(protobuf_CXXFLAGS "-Wno-maybe-uninitialized -Wno-unused-parameter -fPIC -fstack-protector-all -D_FORTIFY_SOURCE=2 -D_GLIBCXX_USE_CXX11_ABI=0 -O2")
  19. set(protobuf_LDFLAGS "-Wl,-z,relro,-z,now,-z,noexecstack")
  20. ExternalProject_Add(protobuf_build
  21. URL ${REQ_URL}
  22. #URL /home/txd/workspace/linux_cmake/pkg/protobuf-3.8.0.tar.gz
  23. #SOURCE_DIR ${GE_CODE_DIR}/../third_party/protobuf/src/protobuf-3.8.0
  24. #DOWNLOAD_COMMAND ${CMAKE_COMMAND} -E copy_directory ${GE_CODE_DIR}/../third_party/protobuf/src/protobuf-3.8.0 <SOURCE_DIR>
  25. #CONFIGURE_COMMAND ${CMAKE_COMMAND}
  26. #-DCMAKE_INSTALL_LIBDIR=${CMAKE_INSTALL_LIBDIR}
  27. #-DCMAKE_C_COMPILER=${CMAKE_C_COMPILER}
  28. #-DCMAKE_CXX_COMPILER=${CMAKE_CXX_COMPILER}
  29. #-DCMAKE_LINKER=${CMAKE_LINKER}
  30. #-DCMAKE_AR=${CMAKE_AR}
  31. #-DCMAKE_RANLIB=${CMAKE_RANLIB}
  32. #-Dprotobuf_WITH_ZLIB=OFF
  33. #-Dprotobuf_BUILD_TESTS=OFF -DBUILD_SHARED_LIBS=ON -DCMAKE_CXX_FLAGS=${protobuf_CXXFLAGS} -DCMAKE_CXX_LDFLAGS=${protobuf_LDFLAGS} -DCMAKE_INSTALL_PREFIX=${CMAKE_INSTALL_PREFIX}/protobuf <SOURCE_DIR>/cmake
  34. CONFIGURE_COMMAND cd <SOURCE_DIR>
  35. && ./autogen.sh && cd <BINARY_DIR> && <SOURCE_DIR>/configure --prefix=${CMAKE_INSTALL_PREFIX}/protobuf --with-zlib=no CC=${CMAKE_C_COMPILER} CXX=${CMAKE_CXX_COMPILER} CXXFLAGS=${protobuf_CXXFLAGS} LDFLAGS=${protobuf_LDFLAGS}
  36. && bash -c "sed -i 's|^hardcode_libdir_flag_spec=.*|hardcode_libdir_flag_spec=\"\"|g' libtool && sed -i 's|^runpath_var=LD_RUN_PATH|runpath_var=DIE_RPATH_DIE|g' libtool"
  37. BUILD_COMMAND $(MAKE)
  38. INSTALL_COMMAND $(MAKE) install
  39. EXCLUDE_FROM_ALL TRUE
  40. )
  41. include(GNUInstallDirs)
  42. set(PROTOBUF_SHARED_PKG_DIR ${CMAKE_INSTALL_PREFIX}/protobuf)
  43. add_library(protobuf SHARED IMPORTED)
  44. file(MAKE_DIRECTORY ${PROTOBUF_SHARED_PKG_DIR}/include)
  45. set_target_properties(protobuf PROPERTIES
  46. IMPORTED_LOCATION ${PROTOBUF_SHARED_PKG_DIR}/lib/libprotobuf.so
  47. )
  48. target_include_directories(protobuf INTERFACE ${PROTOBUF_SHARED_PKG_DIR}/include)
  49. set(INSTALL_BASE_DIR "")
  50. set(INSTALL_LIBRARY_DIR lib)
  51. install(FILES ${PROTOBUF_SHARED_PKG_DIR}/lib/libprotobuf.so ${PROTOBUF_SHARED_PKG_DIR}/lib/libprotobuf.so.19.0.0 OPTIONAL
  52. DESTINATION ${INSTALL_LIBRARY_DIR})
  53. add_dependencies(protobuf protobuf_build)
  54. #set(HAVE_PROTOBUF TRUE CACHE BOOL "protobuf build add")
  55. set(HAVE_PROTOBUF TRUE)

图引擎模块(GE)是MindSpore的一个子模块,其代码由C++实现,位于前端模块ME和底层硬件之间,起到承接作用。图引擎模块以ME下发的图作为输入,然后进行一系列的深度图优化操作,最后输出一张可以在底层硬件上高效运行的图。GE针对昇腾AI处理器的硬件结构特点,做了特定的优化工作,以此来充分发挥出昇腾AI处理器的强大算力。在进行模型训练/推理时,GE会被自动调用而用户并不感知。GE主要由GE API和GE Core两部分组成,详细的架构图如下所示