文档
注册

函数:create_kernel

C函数原型

aclError aclopCreateKernel(const char *opType,

const char *kernelId,

const char *kernelName,

void *binData,

int binSize,

aclopEngineType enginetype,

aclDataDeallocator deallocator)

Python函数

ret = acl.op.create_kernel(op_type, kernel_id, kernel_name, bin_data, bin_size, enginetype, deallocator)

函数功能

动态Shape场景下,将算子注册到系统内部,运行算子时使用。

输入说明

op_type:str,算子类型。

kernel_id:str,算子执行时要指定的Kernel ID。

kernel_name:str,算子Kernel名称,和算子二进制文件中的kernelName保持一致。

bin_data:int,算子Kernel文件的内存地址。

bin_size:int,算子Kernel文件的内存大小,单位为byte。

enginetype:int,表示算子执行引擎,该参数只有acl.op.update.params接口的compile_flag参数值为ACL_COMPILE_SYS时有效。

  • 0:ACL_ENGINE_SYS,不关心具体执行引擎时填写。
  • 1:ACL_ENGINE_AICORE,将算子编译成AI Core算子。
  • 2:ACL_ENGINE_VECTOR,将算子编译成Vector Core算子。

deallocator:int,指定是否自动释放bin_data内存。

  • 0:不自动释放,数据由调用者自行编写代码进行释放。
  • 1:自动释放,内部会设置回调函数释放bin_data内存。

返回值说明

ret:int,错误码。

  • 返回0表示成功。
  • 返回其它值表示失败。

约束说明

参考资源

接口调用流程及示例,参见动态Shape算子(注册算子选择器)

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词