文档
注册
评分
提单
论坛
小AI

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})
搜索结果
找到“0”个结果

当前产品无相关内容

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