CANN/asc-devkit算子IR实例数量声明API

📅 2026/7/18 12:17:00 👁️ 阅读次数 📝 编程学习
CANN/asc-devkit算子IR实例数量声明API

IrInstanceNum

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言,原生支持C和C++标准规范,主要由类库和语言扩展层构成,提供多层级API,满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit

功能说明

基于算子的IR定义,声明实例化时每种输入的实际个数。

函数原型

ContextBuilder &IrInstanceNum(std::vector<uint32_t> instanceNum)

参数说明

参数

输入/输出

说明

instanceNum

输入

基于算子IR原型定义,按照输入的index顺序声明实例化个数。

返回值说明

当前ContextBuilder的对象

约束说明

必须配合NodeIoNum一同使用。

调用示例

context_ascendc::ContextBuilder builder; (void)builder.NodeIoNum(5,3) // 算子有5种输入,3种输出 .IrInstanceNum({1, 1, 3, 1, 1}); // 算子实例化时,index为2的动态类型输入tensor有3个实例

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言,原生支持C和C++标准规范,主要由类库和语言扩展层构成,提供多层级API,满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit

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