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)

参数说明

返回值说明

当前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