注册算子支持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})