news 2026/6/20 5:38:22

CANN/GE RunGraph API文档

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
CANN/GE RunGraph API文档

RunGraph

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

产品支持情况

产品

是否支持

Atlas A3 训练系列产品 / Atlas A3 推理系列产品

Atlas A2 训练系列产品 / Atlas A2 推理系列产品

头文件/库文件

  • 头文件:#include <ge/ge_api_v2.h>
  • 库文件:libge_runner_v2.so

功能说明

同步运行指定ID对应的Graph图,输出运行结果。

本接口与RunGraphAsync/RunGraphWithStreamAsync互斥;若在调用本接口前未执行LoadGraph完成图加载,则本接口将自动调用LoadGraph以完成加载。

函数原型

Status RunGraph(uint32_t graph_id, const std::vector<gert::Tensor>& inputs, std::vector<gert::Tensor>& outputs)

参数说明

参数名

输入/输出

说明

graph_id

输入

要运行图对应的ID。

inputs

输入

计算图输入数据,可以位于Host,也可以位于Device上。

如果数据位于Host上,而执行在device上,用户输入进入数据队列时,需要对每个输入做一次内存的申请和拷贝,输入较大或者较多时可能存在性能瓶颈。

outputs

输出

计算图输出Tensor,用户无需分配内存空间,执行完成后GE会分配Host内存并赋值。

返回值说明

参数名

类型

说明

-

Status

  • SUCCESS:运行子图成功。
  • FAILED:运行子图失败。

约束说明

inputs与图中的data节点相对应,data节点的index属性表征inputs列表中对应数据的位置。即用户需要保证,可以按照data节点的index属性从inputs中获取对应的数据,否则返回错误。如果图中没有data节点,也可以输入的inputs为空。输出的outputs与用户指定的输出节点及输出端口个数与顺序相一致。

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

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/6/20 5:24:48

Dify本地部署构建AI Agent可信评测沙盒实战指南

1. 为什么企业现在必须自己搭一个Agent评测平台——而不是继续用“试用版”或“SaaS界面点几下”最近三个月&#xff0c;我帮六家不同行业的客户做过AI落地可行性评估&#xff0c;其中五家在第二轮沟通时都抛出同一个问题&#xff1a;“你们说的Agent能自动跑测试用例、能调API…

作者头像 李华
网站建设 2026/6/20 5:20:49

低代码表单在企业流程管理中的应用场景

低代码表单在企业流程管理中的应用场景 企业里每天都在产生各种各样的单据&#xff1a;请假、报销、采购、用印、立项、客户拜访、设备报修、入职登记……它们的共同点是“都需要填一张表、走一段流程”。但它们的字段、规则、审批路径又各不相同&#xff0c;而且经常变化。 …

作者头像 李华
网站建设 2026/6/20 5:20:40

Umi-OCR终极指南:三步实现免费离线文字识别与数字提取

Umi-OCR终极指南&#xff1a;三步实现免费离线文字识别与数字提取 【免费下载链接】Umi-OCR OCR software, free and offline. 开源、免费的离线OCR软件。支持截屏/批量导入图片&#xff0c;PDF文档识别&#xff0c;排除水印/页眉页脚&#xff0c;扫描/生成二维码。内置多国语言…

作者头像 李华
网站建设 2026/6/20 5:11:02

AI 工程演化:从 Prompt 到 Loop,未来的方向在哪里

一个工程师让 AI Agent 修一个数据库慢查询。Agent 改了代码&#xff0c;测试挂了&#xff1b;工程师把报错贴回去&#xff0c;Agent 再改&#xff0c;CI 又挂了&#xff1b;工程师再排查再贴&#xff0c;代码审查又没过——六轮之后&#xff0c;工程师花的时间比自己写还多。 …

作者头像 李华