函数：register_compile_func
产品支持情况
|
产品 |
是否支持 |
|---|---|
|
|
√ |
|
|
√ |
|
|
√ |
|
|
√ |
|
|
√ |
|
|
√ |
功能说明
动态Shape场景下，注册算子选择器，用于在算子执行时，能针对不同shape，选择相应的Tiling策略。
如果某算子已注册算子选择器，则不允许重新注册，如果需要变更算子选择器，必须先调用acl.op.unregister_compile_func接口取消注册，然后再调用acl.op.register_compile_func接口重新注册。
函数原型
- C函数原型
1aclError aclopRegisterCompileFunc(const char *opType, aclopCompileFunc func)
- python函数
1ret = acl.op.register_compile_func(op_type, 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表示成功，返回其它值表示失败。 |
父主题： 单算子模型执行