昇腾社区首页
中文
注册

aclgrphSetOpAttr

产品支持情况

产品

是否支持

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

Atlas 200/300/500 推理产品

功能说明

用于支持设置node属性方式的改图。

函数原型

1
graphStatus aclgrphSetOpAttr(Graph &graph, aclgrphAttrType attr_type, const char_t *cfg_path)

参数说明

参数名

输入/输出

说明

graph

输入/输出

作为输入时,代表要通过属性方式修改的图;

作为输出时表示已经修改后的图。

attr_type

输入

attr_type的枚举类型定义:

enum aclgrphAttrType { ATTR_TYPE_KEEP_DTYPE = 0, ATTR_TYPE_WEIGHT_COMPRESS };

ATTR_TYPE_KEEP_DTYPE:保持模型编译时个别算子的计算精度不变。

ATTR_TYPE_WEIGHT_COMPRESS :模型编译时对个别算子进行weight压缩。

cfg_path

输出

配置文件路径。配置文件举例:

计算精度不变的文件格式为:

Opname1
Opname2
…

weight压缩文件格式为:

Opname1;Opname2

返回值说明

参数名

类型

说明

-

graphStatus

0:成功。

其他值:失败。

约束说明