文档
注册

函数:register_compile_func

C函数原型

aclError aclopRegisterCompileFunc(const char *opType, aclopCompileFunc func)

Python函数

ret = acl.op.register_compile_func(op_type, func)

函数功能

动态Shape场景下,注册算子选择器,用于在算子执行时,能针对不同shape,选择相应的Tiling策略。

如果某算子已注册算子选择器,则不允许重新注册,如果需要变更算子选择器,必须先调用acl.op.unregister_compile_func接口取消注册,然后再调用acl.op.register_compile_func接口重新注册。

输入说明

op_type:str,算子类型。

func:python函数对象,算子选择器回调函数,函数定义:

Python侧格式如下:

def call_back_func(num_inputs, input_desc, num_outputs, output_desc, op_attr, aclop_kernel_desc):
pass

返回值说明

ret:int,错误码。

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

约束说明

参考资源

接口调用流程、示例,参见单算子调用

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

当前产品无相关内容

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