文档
注册

--op_debug_list

功能说明

指定要编译的AI Core算子列表。

关联参数

若想使用该参数,则必须和--op_debug_config参数配合使用,算子编译时,只编译该列表指定的算子,并按照--op_debug_config配置文件中的选项进行编译。

参数取值

参数值:算子名称或算子类型。

参数值格式:算子之间使用英文逗号分隔,若为算子类型,则以OpType::typeName格式进行配置,支持算子类型和算子名称混合配置。

参数值约束:要编译的算子,必须放在--op_debug_config参数指定的配置文件中。算子类型必须为基于Ascend IR定义的算子的类型,算子类型查看方法请参见如何确定原始框架网络模型中的算子与昇腾AI处理器支持的算子的对应关系

推荐配置及收益

无。

示例

--op_debug_config参数指定的配置文件(例如gm_debug.cfg)中增加如下信息:

op_debug_config=ccec_O0,ccec_g,oom
op_debug_list=GatherV2,opType::ReduceSum

将该文件上传到ATC工具所在服务器,例如上传到$HOME/module,使用示例如下:

--op_debug_config=$HOME/module/gm_debug.cfg

实际模型转换时,转换命令中无需给出--op_debug_list参数,直接通过--op_debug_config参数读取配置文件中指定的算子列表。

支持的型号

Atlas 200/300/500 推理产品

Atlas 推理系列产品(Ascend 310P处理器)

Atlas 训练系列产品

Atlas 200/500 A2推理产品

Atlas A2训练系列产品

使用约束

无。

搜索结果
找到“0”个结果

当前产品无相关内容

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