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.

gflags.cmake 1.9 kB

4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. if (HAVE_GFLAGS)
  2. return()
  3. endif()
  4. include(ExternalProject)
  5. if ((${CMAKE_INSTALL_PREFIX} STREQUAL /usr/local) OR
  6. (${CMAKE_INSTALL_PREFIX} STREQUAL "C:/Program Files (x86)/ascend"))
  7. set(CMAKE_INSTALL_PREFIX ${GE_CODE_DIR}/output CACHE STRING "path for install()" FORCE)
  8. message(STATUS "No install prefix selected, default to ${CMAKE_INSTALL_PREFIX}.")
  9. endif()
  10. if (GE_PB_PKG)
  11. set(REQ_URL "${GE_PB_PKG}/libs/gflags/v2.2.2.tar.gz")
  12. set(MD5 "1a865b93bacfa963201af3f75b7bd64c")
  13. else()
  14. if (ENABLE_GITEE)
  15. set(REQ_URL "https://gitee.com/mirrors/gflags/repository/archive/v2.2.2.tar.gz")
  16. set(MD5 "")
  17. else()
  18. set(REQ_URL "https://github.com/gflags/gflags/archive/v2.2.2.tar.gz")
  19. set(MD5 "1a865b93bacfa963201af3f75b7bd64c")
  20. endif ()
  21. endif ()
  22. set (gflags_CXXFLAGS "-D_GLIBCXX_USE_CXX11_ABI=0 -Dgoogle=ascend_private")
  23. ExternalProject_Add(gflags_build
  24. URL ${REQ_URL}
  25. #URL /home/txd/workspace/linux_cmake/pkg/protobuf-3.8.0.tar.gz
  26. #SOURCE_DIR ${GE_CODE_DIR}/../../third_party/gflags/src/gflags-2.2.2
  27. TLS_VERIFY OFF
  28. CONFIGURE_COMMAND ${CMAKE_COMMAND} -DCMAKE_CXX_FLAGS=${gflags_CXXFLAGS} -DCMAKE_INSTALL_PREFIX=${CMAKE_INSTALL_PREFIX}/gflags <SOURCE_DIR>
  29. BUILD_COMMAND $(MAKE)
  30. INSTALL_COMMAND $(MAKE) install
  31. EXCLUDE_FROM_ALL TRUE
  32. )
  33. set(GFLAGS_PKG_DIR ${CMAKE_INSTALL_PREFIX}/gflags)
  34. add_library(gflags_static STATIC IMPORTED)
  35. set_target_properties(gflags_static PROPERTIES
  36. IMPORTED_LOCATION ${GFLAGS_PKG_DIR}/lib/libgflags.a
  37. )
  38. add_library(gflags INTERFACE)
  39. target_include_directories(gflags INTERFACE ${GFLAGS_PKG_DIR}/include)
  40. target_link_libraries(gflags INTERFACE gflags_static)
  41. add_dependencies(gflags gflags_build)
  42. #set(HAVE_GFLAGS TRUE CACHE BOOL "gflags build add")
  43. set(HAVE_GFLAGS TRUE)

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