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.

gtest.cmake 2.8 kB

4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081
  1. if (HAVE_GTEST)
  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/ge_gtest/release-1.8.1.tar.gz")
  12. set(MD5 "")
  13. elseif (ENABLE_GITEE)
  14. set(REQ_URL "https://gitee.com/mirrors/googletest/repository/archive/release-1.8.1.tar.gz")
  15. set(MD5 "")
  16. else()
  17. set(REQ_URL "https://github.com/google/googletest/archive/release-1.8.1.tar.gz")
  18. set(MD5 "")
  19. endif ()
  20. set (gtest_CXXFLAGS "-D_GLIBCXX_USE_CXX11_ABI=0 -D_FORTIFY_SOURCE=2 -O2 -fstack-protector-all -Wl,-z,relro,-z,now,-z,noexecstack")
  21. set (gtest_CFLAGS "-D_GLIBCXX_USE_CXX11_ABI=0 -D_FORTIFY_SOURCE=2 -O2 -fstack-protector-all -Wl,-z,relro,-z,now,-z,noexecstack")
  22. ExternalProject_Add(gtest_build
  23. URL ${REQ_URL}
  24. TLS_VERIFY OFF
  25. CONFIGURE_COMMAND ${CMAKE_COMMAND} -DCMAKE_CXX_FLAGS=${gtest_CXXFLAGS} -DCMAKE_INSTALL_PREFIX=${CMAKE_INSTALL_PREFIX}/gtest <SOURCE_DIR>
  26. -DBUILD_TESTING=OFF -DCMAKE_POSITION_INDEPENDENT_CODE=ON -DBUILD_SHARED_LIBS=ON -DCMAKE_MACOSX_RPATH=TRUE
  27. BUILD_COMMAND $(MAKE)
  28. INSTALL_COMMAND $(MAKE) install
  29. EXCLUDE_FROM_ALL TRUE
  30. )
  31. set(GTEST_PKG_DIR ${CMAKE_INSTALL_PREFIX}/gtest)
  32. file(MAKE_DIRECTORY ${GTEST_PKG_DIR}/include)
  33. add_library(gtest SHARED IMPORTED)
  34. set_target_properties(gtest PROPERTIES
  35. IMPORTED_LOCATION ${GTEST_PKG_DIR}/lib/libgtest.so
  36. )
  37. add_library(gtest_main SHARED IMPORTED)
  38. set_target_properties(gtest_main PROPERTIES
  39. IMPORTED_LOCATION ${GTEST_PKG_DIR}/lib/libgtest_main.so
  40. )
  41. target_include_directories(gtest INTERFACE ${GTEST_PKG_DIR}/include)
  42. target_include_directories(gtest_main INTERFACE ${GTEST_PKG_DIR}/include)
  43. add_library(gmock SHARED IMPORTED)
  44. set_target_properties(gmock PROPERTIES
  45. IMPORTED_LOCATION ${GTEST_PKG_DIR}/lib/libgmock.so
  46. )
  47. add_library(gmock_main SHARED IMPORTED)
  48. set_target_properties(gmock_main PROPERTIES
  49. IMPORTED_LOCATION ${GTEST_PKG_DIR}/lib/libgmock_main.so
  50. )
  51. target_include_directories(gmock INTERFACE ${GTEST_PKG_DIR}/include)
  52. target_include_directories(gmock_main INTERFACE ${GTEST_PKG_DIR}/include)
  53. set(INSTALL_BASE_DIR "")
  54. set(INSTALL_LIBRARY_DIR lib)
  55. install(FILES ${GTEST_PKG_DIR}/lib/libgtest.so ${GTEST_PKG_DIR}/lib/libgtest_main.so ${GTEST_PKG_DIR}/lib/libgmock.so ${GTEST_PKG_DIR}/lib/libgmock_main.so OPTIONAL
  56. DESTINATION ${INSTALL_LIBRARY_DIR})
  57. add_dependencies(gtest gtest_build)
  58. #set(HAVE_GFLAGS TRUE CACHE BOOL "gflags build add")
  59. set(HAVE_GTEST TRUE)

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