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_static.cmake 2.9 kB

4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. include(ExternalProject)
  2. include(GNUInstallDirs)
  3. #set(CMAKE_INSTALL_PREFIX ${GE_CODE_DIR}/output)
  4. if ((${CMAKE_INSTALL_PREFIX} STREQUAL /usr/local) OR
  5. (${CMAKE_INSTALL_PREFIX} STREQUAL "C:/Program Files (x86)/ascend"))
  6. set(CMAKE_INSTALL_PREFIX ${GE_CODE_DIR}/output CACHE STRING "path for install()" FORCE)
  7. message(STATUS "No install prefix selected, default to ${CMAKE_INSTALL_PREFIX}.")
  8. endif()
  9. if(GE_PB_PKG)
  10. set(REQ_URL "${GE_PB_PKG}/libs/protobuf/v3.8.0.tar.gz")
  11. else()
  12. if (ENABLE_GITEE)
  13. set(REQ_URL "https://gitee.com/mirrors/protobuf_source/repository/archive/v3.8.0.tar.gz")
  14. set(MD5 "eba86ae9f07ba5cfbaf8af3bc4e84236")
  15. else()
  16. set(REQ_URL "https://github.com/protocolbuffers/protobuf/archive/v3.8.0.tar.gz")
  17. set(MD5 "3d9e32700639618a4d2d342c99d4507a")
  18. endif ()
  19. endif()
  20. set(protobuf_CXXFLAGS "-Wno-maybe-uninitialized -Wno-unused-parameter -fPIC -fstack-protector-all -D_FORTIFY_SOURCE=2 -D_GLIBCXX_USE_CXX11_ABI=0 -O2 -Dgoogle=ascend_private")
  21. set(protobuf_LDFLAGS "-Wl,-z,relro,-z,now,-z,noexecstack")
  22. set(PROTOBUF_STATIC_PKG_DIR ${CMAKE_INSTALL_PREFIX}/protobuf_static)
  23. ExternalProject_Add(protobuf_static_build
  24. URL ${REQ_URL}
  25. #URL /home/txd/workspace/linux_cmake/pkg/protobuf-3.8.0.tar.gz
  26. #SOURCE_DIR ${METADEF_DIR}/../../third_party/protobuf/src/protobuf-3.8.0
  27. TLS_VERIFY OFF
  28. CONFIGURE_COMMAND ${CMAKE_COMMAND}
  29. -DCMAKE_C_COMPILER=${CMAKE_C_COMPILER}
  30. -DCMAKE_CXX_COMPILER=${CMAKE_CXX_COMPILER}
  31. -DCMAKE_INSTALL_LIBDIR=${CMAKE_INSTALL_LIBDIR}
  32. -DCMAKE_LINKER=${CMAKE_LINKER}
  33. -DCMAKE_AR=${CMAKE_AR}
  34. -DCMAKE_RANLIB=${CMAKE_RANLIB}
  35. -Dprotobuf_WITH_ZLIB=OFF
  36. -DLIB_PREFIX=ascend_
  37. -Dprotobuf_BUILD_TESTS=OFF -DCMAKE_CXX_FLAGS=${protobuf_CXXFLAGS} -DCMAKE_CXX_LDFLAGS=${protobuf_LDFLAGS} -DCMAKE_INSTALL_PREFIX=${PROTOBUF_STATIC_PKG_DIR} <SOURCE_DIR>/cmake
  38. BUILD_COMMAND $(MAKE)
  39. INSTALL_COMMAND $(MAKE) install
  40. EXCLUDE_FROM_ALL TRUE
  41. )
  42. include(GNUInstallDirs)
  43. add_library(ascend_protobuf_static_lib STATIC IMPORTED)
  44. set_target_properties(ascend_protobuf_static_lib PROPERTIES
  45. IMPORTED_LOCATION ${PROTOBUF_STATIC_PKG_DIR}/${CMAKE_INSTALL_LIBDIR}/libascend_protobuf.a
  46. )
  47. add_library(ascend_protobuf_static INTERFACE)
  48. target_include_directories(ascend_protobuf_static INTERFACE ${PROTOBUF_STATIC_PKG_DIR}/include)
  49. target_link_libraries(ascend_protobuf_static INTERFACE ascend_protobuf_static_lib)
  50. if (ENABLE_D OR ENABLE_ACL OR ENABLE_MS_TESTCASES)
  51. include_directories(${PROTOBUF_STATIC_PKG_DIR}/include)
  52. endif ()
  53. add_dependencies(ascend_protobuf_static protobuf_static_build)

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