昇腾社区首页
中文
注册

aclopCreateKernel

产品支持情况

产品

是否支持

Atlas A3 训练系列产品 / Atlas A3 推理系列产品

Atlas A2 训练系列产品 / Atlas 800I A2 推理产品 /A200I A2 Box 异构组件

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

Atlas 200/300/500 推理产品

功能说明

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

函数原型

aclError aclopCreateKernel(const char *opType,
const char *kernelId,
const char *kernelName,
void *binData,
int binSize,
aclopEngineType enginetype,
aclDataDeallocator deallocator)

参数说明

参数名

输入/输出

说明

opType

输入

算子类型的指针。

kernelId

输入

算子执行时要指定的Kernel ID的指针。

kernelName

输入

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

binData

输入

算子Kernel文件的内存地址指针。

binSize

输入

算子Kernel文件的内存大小,单位为Byte。

enginetype

输入

表示算子执行引擎。

deallocator

输入

释放binData内存的回调函数,调用者根据binData的构造方法设置对应的释放函数。如果数据由调用者释放则设置为空。

函数签名为:

typedef void (*aclDataDeallocator)(void *data, size_t length);

返回值说明

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

参考资源

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