Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
taoxiangdong 07f5327b18 | 3 years ago | |
---|---|---|
.. | ||
Makefile | 4 years ago | |
README | 4 years ago | |
README.md | 4 years ago | |
gen_stubapi.py | 3 years ago |
Makefile:
'''
ATC_INCLUDE_DIR := $(ASCEND_PATH)/atc/include
OPP_INCLUDE_DIR := $(ASCEND_PATH)/opp/op_proto/built-in/inc
LOCAL_MODULE_NAME := ir_build
CC := g++
CFLAGS := -std=c++11 -g -Wall
SRCS := $(wildcard $(LOCAL_DIR)/main.cpp)
INCLUDES := -I $(ASCEND_OPP_PATH)/op_proto/built-in/inc
-I $(ATC_INCLUDE_DIR)/graph
-I $(ATC_INCLUDE_DIR)/ge \
LIBS := -L ${ASCEND_PATH}/atc/lib64/stub
-lgraph
-lge_compiler
ir_build:
mkdir -p out
$(CC) $(SRCS) $(INCLUDES) $(LIBS) $(CFLAGS) -o ./out/$(LOCAL_MODULE_NAME)
clean:
rm -rf out
'''
make
export LD_LIBRARY_PATH= $(ASCEND_PATH)/atc/lib64
图引擎模块(GE)是MindSpore的一个子模块,其代码由C++实现,位于前端模块ME和底层硬件之间,起到承接作用。图引擎模块以ME下发的图作为输入,然后进行一系列的深度图优化操作,最后输出一张可以在底层硬件上高效运行的图。GE针对昇腾AI处理器的硬件结构特点,做了特定的优化工作,以此来充分发挥出昇腾AI处理器的强大算力。在进行模型训练/推理时,GE会被自动调用而用户并不感知。GE主要由GE API和GE Core两部分组成,详细的架构图如下所示
C++ Text Makefile Shell CMake other