CANN/asc-devkit算子参数校验标志

📅 2026/7/18 9:14:08 👁️ 阅读次数 📝 编程学习
CANN/asc-devkit算子参数校验标志

NeedCheckSupportFlag

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

功能说明

标识是否在算子融合阶段调用算子参数校验函数进行data type与shape的校验。

  • 若配置为"true",框架会调用通过SetCheckSupport设置的算子参数校验函数,检查算子是否支持指定输入,此场景下需要自行实现算子参数校验的回调函数。
  • 若配置为"false",表示不需要进行校验。

函数原型

OpAICoreConfig &NeedCheckSupportFlag(bool flag)

参数说明

参数

输入/输出

说明

flag

输入

标识是否在算子融合阶段调用算子参数校验函数进行data type与shape的校验。

返回值说明

OpAICoreConfig类,请参考OpAICoreConfig。

约束说明

调用示例

请参考SetCheckSupport节调用示例。

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

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