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.

protoc.cmake 4.8 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112
  1. if (HAVE_PROTOC)
  2. return()
  3. endif()
  4. include(ExternalProject)
  5. include(GNUInstallDirs)
  6. #set(CMAKE_INSTALL_PREFIX ${GE_CODE_DIR}/output)
  7. if ((${CMAKE_INSTALL_PREFIX} STREQUAL /usr/local) OR
  8. (${CMAKE_INSTALL_PREFIX} STREQUAL "C:/Program Files (x86)/ascend"))
  9. set(CMAKE_INSTALL_PREFIX ${GE_CODE_DIR}/output CACHE STRING "path for install()" FORCE)
  10. message(STATUS "No install prefix selected, default to ${CMAKE_INSTALL_PREFIX}.")
  11. endif()
  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. set(protobuf_CXXFLAGS "-Wno-maybe-uninitialized -Wno-unused-parameter -fPIC -fstack-protector-all -D_FORTIFY_SOURCE=2 -D_GLIBCXX_USE_CXX11_ABI=0 -O2")
  20. set(protobuf_LDFLAGS "-Wl,-z,relro,-z,now,-z,noexecstack")
  21. ExternalProject_Add(protoc_build
  22. URL ${REQ_URL}
  23. #URL /home/txd/workspace/linux_cmake/pkg/protobuf-3.8.0.tar.gz
  24. #SOURCE_DIR ${GE_CODE_DIR}/../third_party/protobuf/src/protobuf-3.8.0
  25. CONFIGURE_COMMAND ${CMAKE_COMMAND} -Dprotobuf_WITH_ZLIB=OFF -Dprotobuf_BUILD_TESTS=OFF -DBUILD_SHARED_LIBS=OFF -DCMAKE_CXX_FLAGS=${protobuf_CXXFLAGS} -DCMAKE_CXX_LDFLAGS=${protobuf_LDFLAGS} -DCMAKE_INSTALL_PREFIX=${CMAKE_INSTALL_PREFIX}/protoc <SOURCE_DIR>/cmake
  26. BUILD_COMMAND $(MAKE)
  27. INSTALL_COMMAND $(MAKE) install
  28. EXCLUDE_FROM_ALL TRUE
  29. )
  30. set(PROTOC_PKG_DIR ${CMAKE_INSTALL_PREFIX}/protoc)
  31. set(protoc_EXECUTABLE ${PROTOC_PKG_DIR}/${CMAKE_INSTALL_BINDIR}/protoc)
  32. function(protobuf_generate comp c_var h_var)
  33. if(NOT ARGN)
  34. message(SEND_ERROR "Error: protobuf_generate() called without any proto files")
  35. return()
  36. endif()
  37. set(${c_var})
  38. set(${h_var})
  39. foreach(file ${ARGN})
  40. get_filename_component(abs_file ${file} ABSOLUTE)
  41. get_filename_component(file_name ${file} NAME_WE)
  42. get_filename_component(file_dir ${abs_file} PATH)
  43. get_filename_component(parent_subdir ${file_dir} NAME)
  44. if("${parent_subdir}" STREQUAL "proto")
  45. set(proto_output_path ${CMAKE_BINARY_DIR}/proto/${comp}/proto)
  46. else()
  47. set(proto_output_path ${CMAKE_BINARY_DIR}/proto/${comp}/proto/${parent_subdir})
  48. endif()
  49. list(APPEND ${c_var} "${proto_output_path}/${file_name}.pb.cc")
  50. list(APPEND ${h_var} "${proto_output_path}/${file_name}.pb.h")
  51. add_custom_command(
  52. OUTPUT "${proto_output_path}/${file_name}.pb.cc" "${proto_output_path}/${file_name}.pb.h"
  53. WORKING_DIRECTORY ${PROJECT_SOURCE_DIR}
  54. COMMAND ${CMAKE_COMMAND} -E make_directory "${proto_output_path}"
  55. COMMAND ${protoc_EXECUTABLE} -I${file_dir} --cpp_out=${proto_output_path} ${abs_file}
  56. DEPENDS protoc_build ${abs_file}
  57. COMMENT "Running C++ protocol buffer compiler on ${file}" VERBATIM )
  58. endforeach()
  59. set_source_files_properties(${${c_var}} ${${h_var}} PROPERTIES GENERATED TRUE)
  60. set(${c_var} ${${c_var}} PARENT_SCOPE)
  61. set(${h_var} ${${h_var}} PARENT_SCOPE)
  62. endfunction()
  63. function(protobuf_generate_py comp py_var)
  64. if(NOT ARGN)
  65. message(SEND_ERROR "Error: protobuf_generate_py() called without any proto files")
  66. return()
  67. endif()
  68. set(${py_var})
  69. foreach(file ${ARGN})
  70. get_filename_component(abs_file ${file} ABSOLUTE)
  71. get_filename_component(file_name ${file} NAME_WE)
  72. get_filename_component(file_dir ${abs_file} PATH)
  73. get_filename_component(parent_subdir ${file_dir} NAME)
  74. if("${parent_subdir}" STREQUAL "proto")
  75. set(proto_output_path ${CMAKE_BINARY_DIR}/proto/${comp}/proto)
  76. else()
  77. set(proto_output_path ${CMAKE_BINARY_DIR}/proto/${comp}/proto/${parent_subdir})
  78. endif()
  79. list(APPEND ${py_var} "${proto_output_path}/${file_name}_pb2.py")
  80. add_custom_command(
  81. OUTPUT "${proto_output_path}/${file_name}_pb2.py"
  82. WORKING_DIRECTORY ${PROJECT_SOURCE_DIR}
  83. COMMAND ${CMAKE_COMMAND} -E make_directory "${proto_output_path}"
  84. COMMAND ${protoc_EXECUTABLE} -I${file_dir} --python_out=${proto_output_path} ${abs_file}
  85. DEPENDS protoc_build ${abs_file}
  86. COMMENT "Running PYTHON protocol buffer compiler on ${file}" VERBATIM )
  87. endforeach()
  88. set_source_files_properties(${${py_var}} PROPERTIES GENERATED TRUE)
  89. set(${py_var} ${${py_var}} PARENT_SCOPE)
  90. endfunction()
  91. #set(HAVE_PROTOC TRUE CACHE BOOL "protoc build add")
  92. set(HAVE_PROTOC TRUE)

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