d5b140e5创建于 2025年12月26日历史提交

aclgrphConvertToWeightRefreshableGraphs

产品支持情况

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√

头文件/库文件

头文件：#include <ge/ge_ir_build.h>
库文件：libge_compiler.so

功能说明

通过传入Const节点名数组将原图转换成一组权重可更新的图。

该接口适用于权重更新场景。

函数原型

graphStatus aclgrphConvertToWeightRefreshableGraphs(const ge::Graph &origin_graph, const std::vector<AscendString> &const_names, WeightRefreshableGraphs &weight_refreshable_graphs)

参数说明

参数名	输入/输出	说明
origin_graph	输入	需要权重更新的原图。
const_names	输入	需要权重更新的节点名数组。
weight_refreshable_graphs	输出	权重可更新的图，该参数为一结构体，包括三部分：权重初始化图，权重更新图，推理图。 struct WeightRefreshableGraphs { ge::Graph infer_graph; ge::Graph var_init_graph; ge::Graph var_update_graph; };

参数名

输入/输出

说明

origin_graph

输入

需要权重更新的原图。

const_names

输入

需要权重更新的节点名数组。

weight_refreshable_graphs

输出

权重可更新的图，该参数为一结构体，包括三部分：权重初始化图，权重更新图，推理图。

struct WeightRefreshableGraphs {
  ge::Graph infer_graph;
  ge::Graph var_init_graph;
  ge::Graph var_update_graph;
};

返回值说明

参数名	类型	说明
-	graphStatus	GRAPH_SUCCESS(0)：成功。其他值：失败。

参数名

类型

说明

graphStatus

GRAPH_SUCCESS(0)：成功。

其他值：失败。

调用示例

完整调用示例请参见调用示例。