news 2026/9/10 16:06:11

CANN/ge:从内存加载队列化模型

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
CANN/ge:从内存加载队列化模型

aclmdlLoadFromMemWithQ

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

产品支持情况

  • Ascend 950PR/Ascend 950DT:支持
  • Atlas A3 训练系列产品/Atlas A3 推理系列产品:支持
  • Atlas A2 训练系列产品/Atlas A2 推理系列产品:支持
  • Atlas 200I/500 A2 推理产品:支持
  • Atlas 推理系列产品:支持
  • Atlas 训练系列产品:支持
  • IPV350:不支持

功能说明

从内存加载om模型文件数据,模型的输入、输出数据都存放在队列中。本接口只支持加载固定Shape输入的模型。

关于如何获取om模型文件,请参见《ATC离线模型编译工具》中的“参数说明 > 基础功能参数 > 总体选项 > --mode”。

函数原型

aclError aclmdlLoadFromMemWithQ(const void *model, size_t modelSize, uint32_t *modelId, const uint32_t *inputQ, size_t inputQNum, const uint32_t *outputQ, size_t outputQNum)

参数说明

参数名输入/输出说明
model输入存放模型数据的内存地址指针。
modelSize输入内存中的模型数据长度,单位Byte。
modelId输出模型ID的指针。
系统成功加载模型后,返回模型ID作为后续操作时识别模型的标志。
inputQ输入队列ID的指针,一个模型的输入对应一个队列ID。
inputQNum输入输入队列大小。
outputQ输入队列ID的指针,一个模型的输出对应一个队列ID。
outputQNum输入输出队列大小。

返回值说明

返回0表示成功,返回其他值表示失败,请参见aclError。

参考资源

当前还提供了aclmdlSetConfigOpt接口、aclmdlLoadWithConfig接口来实现模型加载,通过配置对象中的属性来区分,在加载模型时是从文件加载,还是从内存加载,以及内存是由系统内部管理,还是由用户管理。

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

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

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

JAVA计算机毕设之基于 Vue+SpringBoot 框架的股票模拟交易教学系统的设计与开发(完整前后端代码+说明文档+LW,调试定制等)

博主介绍:✌️码农一枚 ,专注于大学生项目实战开发、讲解和毕业🚢文撰写修改等。全栈领域优质创作者,博客之星、掘金/华为云/阿里云/InfoQ等平台优质作者、专注于Java、小程序技术领域和毕业项目实战 ✌️技术范围:&am…

作者头像 李华
网站建设 2026/9/10 16:05:40

000001 基本面分析数据

000001 基本面分析数据 【免费下载链接】TradingAgents-CN 基于多智能体LLM的中文金融交易框架 - TradingAgents中文增强版 项目地址: https://gitcode.com/GitHub_Trending/tr/TradingAgents-CN 股票类型: 中国A股 货币: 人民币 () 分析日期: 2025-11-04 数据深度级别:…

作者头像 李华
网站建设 2026/9/10 16:05:36

工业喷码缺陷检测:模板比对+结构校验+语义容错闭环方案

简介:本资源是一套面向工业视觉初学者与自动化质检工程师的OCR喷码缺陷检测实战项目,聚焦于生产线上喷码模糊、缺失、错位等典型缺陷的自动识别与判定问题。压缩包共207个文件,含55张标注样本图(jpg)、26张可视化结果图…

作者头像 李华
网站建设 2026/9/10 16:05:25

Spring WebFlux缓冲区溢出问题解决方案

1. 问题现象与背景解析当你在使用Spring WebFlux或类似响应式编程框架时,可能会遇到这样的错误提示:"Exceeded limit on max bytes to buffer : 262144"。这个报错通常发生在处理HTTP请求体时,特别是当接收的数据量超过框架默认设置…

作者头像 李华
网站建设 2026/9/10 16:05:17

京东云部署OpenClaw:AI应用框架快速接入指南

1. OpenClaw部署前的认知准备OpenClaw作为一款新兴的AI应用框架,其核心价值在于为开发者提供了快速接入多种大语言模型的统一接口。不同于传统的单一模型部署方案,OpenClaw采用了模块化设计理念,允许用户根据实际需求灵活组合不同的AI能力模块…

作者头像 李华
网站建设 2026/9/10 16:05:07

蓝牙Mesh组网实战:从节点原理到配网调试全攻略

蓝牙Mesh这几年被问得越来越多,尤其是从HC-05/06这些经典蓝牙模块入门的老哥们,一听说“网状网络”就以为是“几个设备互相传文件”,等真去翻协议文档,又被一堆“节点、元素、模型、密钥”给整懵。这版指南就是把之前踩过的坑、补…

作者头像 李华