CANN/GE添加图API

AddGraph

【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge

产品支持情况

头文件/库文件

• 头文件：#include <ge/ge_api.h>
• 库文件：libge_runner.so

功能说明

向Session中添加Graph，Session内会生成唯一的Graph ID。

函数原型

[!NOTE]说明 数据类型为string的接口后续版本会废弃，建议使用数据类型为非string的接口。

Status AddGraph(uint32_t graph_id, const Graph &graph) Status AddGraph(uint32_t graph_id, const Graph &graph, const std::map<std::string, std::string> &options) Status AddGraph(uint32_t graph_id, const Graph &graph, const std::map<AscendString, AscendString> &options)

参数说明

返回值说明

约束说明

• 相同对象的Graph调用此接口注册，会导致不同的Graph ID实际共享同一个Graph对象，导致后续操作相互影响而出错。
• 不同的Graph对象请不要使用相同的Graph ID来添加，该情况下，只保留第一次添加的Graph对象，后续的Graph对象不会添加成功。
• 使用该接口，Session会直接修改添加的Graph对象。如果AddGraph后需要保持原有的Graph对象不受影响，应使用AddGraphWithCopy接口，AddGraphWithCopy会在Session中拷贝一份Graph对象，仅对Graph对象的拷贝进行修改。

【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge