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.

README.md 1.1 kB

4 years ago
1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. # "stub" usage:
  2. ## Description
  3. - File libge_compiler.so ,libgraph.so are used in IR build application interface.
  4. # Attention
  5. - Don't link other library except libge_compiler.so ,libgraph.so, as they may be changed in the future.
  6. # Usage
  7. ## Compile: compile the application invoking the IR build API.
  8. Makefile:
  9. '''
  10. ATC_INCLUDE_DIR := $(ASCEND_PATH)/atc/include
  11. OPP_INCLUDE_DIR := $(ASCEND_PATH)/opp/op_proto/built-in/inc
  12. LOCAL_MODULE_NAME := ir_build
  13. CC := g++
  14. CFLAGS := -std=c++11 -g -Wall
  15. SRCS := $(wildcard $(LOCAL_DIR)/main.cpp)
  16. INCLUDES := -I $(ASCEND_OPP_PATH)/op_proto/built-in/inc \
  17. -I $(ATC_INCLUDE_DIR)/graph \
  18. -I $(ATC_INCLUDE_DIR)/ge \
  19. LIBS := -L ${ASCEND_PATH}/atc/lib64/stub \
  20. -lgraph \
  21. -lge_compiler
  22. ir_build:
  23. mkdir -p out
  24. $(CC) $(SRCS) $(INCLUDES) $(LIBS) $(CFLAGS) -o ./out/$(LOCAL_MODULE_NAME)
  25. clean:
  26. rm -rf out
  27. '''
  28. make
  29. ## Run the application after set the LD_LIBRARY_PATH to include the real path of the library which locates in the directory of atc/lib64
  30. export LD_LIBRARY_PATH= $(ASCEND_PATH)/atc/lib64
  31. - ./ ir_build

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