函数: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,错误码。
|
约束说明 |
无 |
注意事项 |
无 |
参考资源 |
接口调用流程、示例,参见单算子调用。 |
父主题: 算子编译