set_fusion_buildcfg
函数功能
注册算子支持compute融合的场景下设置编译选项。
函数原型
def set_fusion_buildcfg(op_type, config)
参数说明
参数名 |
类型 |
说明 |
---|---|---|
op_type |
string |
算子类型 |
config |
dict |
可以设置所有的编译选项。 针对DSL算子,取值范围详见build中的编译选项参数说明。 |
返回值说明
无。
约束说明
此接口仅适用于使用DSL方式开发的算子。
调用示例
tbe.common.register.set_fusion_buildcfg("abs", {"bool_storage_as_1bit": False})
父主题: TBE Register API