hummingbird
/
graphengine
Not watched
2
0
Fork 0
Code
Releases 13 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
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.
283 Commits
28 Branches
21 MB
5223 Download
Tree: 5b5047909c
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '5b5047909c'
${ noResults }
graphengine/ge/graph/build/memory/CMakeLists.txt

CMakeLists.txt 950 B
Raw Normal View History

update cmake
4 years ago
​
​
​
​
​
​
​
​
initial version
5 years ago
update cmake
4 years ago
​
initial version
5 years ago
update cmake
4 years ago
​
​
​
initial version
5 years ago
modified: cmake/external_libs/gflags.cmake modified: cmake/external_libs/protobuf_shared.cmake modified: cmake/external_libs/protobuf_static.cmake modified: ge/CMakeLists.txt modified: ge/client/module.mk modified: ge/common/CMakeLists.txt modified: ge/common/ge_common.mk modified: ge/executor/CMakeLists.txt modified: ge/executor/module.mk modified: ge/ge_inference.mk modified: ge/ge_local_engine/CMakeLists.txt modified: ge/ge_local_engine/module.mk modified: ge/ge_runner.mk modified: ge/graph/build/memory/CMakeLists.txt modified: ge/graph/build/memory/module.mk modified: ge/host_cpu_engine/CMakeLists.txt modified: ge/host_cpu_engine/module.mk modified: ge/offline/CMakeLists.txt modified: ge/offline/module.mk
4 years ago
​
​
​
update cmake
4 years ago
​
modified: cmake/external_libs/gflags.cmake modified: cmake/external_libs/protobuf_shared.cmake modified: cmake/external_libs/protobuf_static.cmake modified: ge/CMakeLists.txt modified: ge/client/module.mk modified: ge/common/CMakeLists.txt modified: ge/common/ge_common.mk modified: ge/executor/CMakeLists.txt modified: ge/executor/module.mk modified: ge/ge_inference.mk modified: ge/ge_local_engine/CMakeLists.txt modified: ge/ge_local_engine/module.mk modified: ge/ge_runner.mk modified: ge/graph/build/memory/CMakeLists.txt modified: ge/graph/build/memory/module.mk modified: ge/host_cpu_engine/CMakeLists.txt modified: ge/host_cpu_engine/module.mk modified: ge/offline/CMakeLists.txt modified: ge/offline/module.mk
4 years ago
update cmake
4 years ago
​
​
​
​
​
​
​
​
​
​
​
​
​
update blue include behind the yellow zone
4 years ago
​
update cmake
4 years ago
 123456789101112131415161718192021222324252627282930313233343536373839404142 
  1. set(SRC_LIST

  2.     "memory_assigner.cc"

  3.     "graph_mem_assigner.cc"

  4.     "binary_block_mem_assigner.cc"

  5.     "block_mem_assigner.cc"

  6.     "hybrid_mem_assigner.cc"

  7.     "max_block_mem_assigner.cc"

  8.     "var_mem_assign_util.cc"

  9. )

  10. 

  11. ############ libge_memory.a ############

  12. add_library(ge_memory STATIC ${SRC_LIST})

  13. 

  14. target_compile_options(ge_memory PRIVATE

  15.     -Werror

  16.     -O2

  17. )

  18. 

  19. target_compile_definitions(ge_memory PRIVATE

  20.     google=ascend_private

  21. )

  22. 

  23. target_link_libraries(ge_memory PRIVATE

  24.     $<BUILD_INTERFACE:intf_pub>

  25.     ascend_protobuf

  26.     c_sec

  27. )

  28. 

  29. target_include_directories(ge_memory PRIVATE

  30.     ${CMAKE_CURRENT_LIST_DIR}

  31.     ${GE_CODE_DIR}/ge

  32.     ${GE_CODE_DIR}/inc

  33.     ${GE_CODE_DIR}/inc/external

  34.     ${METADEF_DIR}/inc

  35.     ${METADEF_DIR}/inc/external

  36.     ${METADEF_DIR}/inc/external/graph

  37.     ${GE_CODE_DIR}/inc/framework

  38.     #### yellow zone ####

  39.     ${GE_CODE_DIR}/../inc

  40.     #### blue zone ####

  41.     ${GE_CODE_DIR}/third_party/fwkacllib/inc

  42. )

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