Star0

Star0

cann-robotfeat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付

5eaca232Add on 15 days agoHistory

Files	Last commit	Last update
examples	feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Co-authored-by: Dazhou_Cao<caodazhou@huawei.com> # message auto-generated for no-merge-commit merge: !6621 merge migrate-gelocal-delivery into master feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Created-by: Dazhou_Cao Commit-by: Dazhou_Cao Merged-by: cann-robot Description: ## 描述汇总迁移 gelocal（RT2.0）算子交付件： - 基础/GE 控制流算子：size、rank、LoopCond、NextIteration、RefIdentity、SwitchByIndex、aipp_data、End、squeezev2、unsqueezev2，以及 enter/exit/merge/switch/switch_n/ref_* 等 - 子图控制流算子：If、Case、For、PartitionedCall、While（原型注册 + RT2.0 推导注册 + UT + README，If/Case/For/PartitionedCall 含 GeIR 示例；均归入 ge 分类） - TF 控制流插件：If/_If/StatelessIf、While/_While/StatelessWhile、Case/StatelessCase、PartitionedCall/StatefulPartitionedCall 变体合并映射 - reshape：infershape 对齐 RT1.0 语义、allowzero/shape 属性注册、信息库 - 文档：README 产品支持表统一（全部√）、gather_shapes 归入 conversion、op_list 同步登记 ## 测试 - `bash build.sh --ophost --opgraph -j8` 构建通过（含各算子 REG_OP 注册校验） - 宿主侧全量 UT：2594/2594 PASSED（384 test suites） ## 说明 - 内容汇总自此前迁移 PR（6474 / 6522 / 6208）的交付件，本 PR 为其合并后的统一交付 See merge request: cann/ops-math!6621	15 days ago
op_graph	feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Co-authored-by: Dazhou_Cao<caodazhou@huawei.com> # message auto-generated for no-merge-commit merge: !6621 merge migrate-gelocal-delivery into master feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Created-by: Dazhou_Cao Commit-by: Dazhou_Cao Merged-by: cann-robot Description: ## 描述汇总迁移 gelocal（RT2.0）算子交付件： - 基础/GE 控制流算子：size、rank、LoopCond、NextIteration、RefIdentity、SwitchByIndex、aipp_data、End、squeezev2、unsqueezev2，以及 enter/exit/merge/switch/switch_n/ref_* 等 - 子图控制流算子：If、Case、For、PartitionedCall、While（原型注册 + RT2.0 推导注册 + UT + README，If/Case/For/PartitionedCall 含 GeIR 示例；均归入 ge 分类） - TF 控制流插件：If/_If/StatelessIf、While/_While/StatelessWhile、Case/StatelessCase、PartitionedCall/StatefulPartitionedCall 变体合并映射 - reshape：infershape 对齐 RT1.0 语义、allowzero/shape 属性注册、信息库 - 文档：README 产品支持表统一（全部√）、gather_shapes 归入 conversion、op_list 同步登记 ## 测试 - `bash build.sh --ophost --opgraph -j8` 构建通过（含各算子 REG_OP 注册校验） - 宿主侧全量 UT：2594/2594 PASSED（384 test suites） ## 说明 - 内容汇总自此前迁移 PR（6474 / 6522 / 6208）的交付件，本 PR 为其合并后的统一交付 See merge request: cann/ops-math!6621	15 days ago
op_host	feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Co-authored-by: Dazhou_Cao<caodazhou@huawei.com> # message auto-generated for no-merge-commit merge: !6621 merge migrate-gelocal-delivery into master feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Created-by: Dazhou_Cao Commit-by: Dazhou_Cao Merged-by: cann-robot Description: ## 描述汇总迁移 gelocal（RT2.0）算子交付件： - 基础/GE 控制流算子：size、rank、LoopCond、NextIteration、RefIdentity、SwitchByIndex、aipp_data、End、squeezev2、unsqueezev2，以及 enter/exit/merge/switch/switch_n/ref_* 等 - 子图控制流算子：If、Case、For、PartitionedCall、While（原型注册 + RT2.0 推导注册 + UT + README，If/Case/For/PartitionedCall 含 GeIR 示例；均归入 ge 分类） - TF 控制流插件：If/_If/StatelessIf、While/_While/StatelessWhile、Case/StatelessCase、PartitionedCall/StatefulPartitionedCall 变体合并映射 - reshape：infershape 对齐 RT1.0 语义、allowzero/shape 属性注册、信息库 - 文档：README 产品支持表统一（全部√）、gather_shapes 归入 conversion、op_list 同步登记 ## 测试 - `bash build.sh --ophost --opgraph -j8` 构建通过（含各算子 REG_OP 注册校验） - 宿主侧全量 UT：2594/2594 PASSED（384 test suites） ## 说明 - 内容汇总自此前迁移 PR（6474 / 6522 / 6208）的交付件，本 PR 为其合并后的统一交付 See merge request: cann/ops-math!6621	15 days ago
tests	feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Co-authored-by: Dazhou_Cao<caodazhou@huawei.com> # message auto-generated for no-merge-commit merge: !6621 merge migrate-gelocal-delivery into master feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Created-by: Dazhou_Cao Commit-by: Dazhou_Cao Merged-by: cann-robot Description: ## 描述汇总迁移 gelocal（RT2.0）算子交付件： - 基础/GE 控制流算子：size、rank、LoopCond、NextIteration、RefIdentity、SwitchByIndex、aipp_data、End、squeezev2、unsqueezev2，以及 enter/exit/merge/switch/switch_n/ref_* 等 - 子图控制流算子：If、Case、For、PartitionedCall、While（原型注册 + RT2.0 推导注册 + UT + README，If/Case/For/PartitionedCall 含 GeIR 示例；均归入 ge 分类） - TF 控制流插件：If/_If/StatelessIf、While/_While/StatelessWhile、Case/StatelessCase、PartitionedCall/StatefulPartitionedCall 变体合并映射 - reshape：infershape 对齐 RT1.0 语义、allowzero/shape 属性注册、信息库 - 文档：README 产品支持表统一（全部√）、gather_shapes 归入 conversion、op_list 同步登记 ## 测试 - `bash build.sh --ophost --opgraph -j8` 构建通过（含各算子 REG_OP 注册校验） - 宿主侧全量 UT：2594/2594 PASSED（384 test suites） ## 说明 - 内容汇总自此前迁移 PR（6474 / 6522 / 6208）的交付件，本 PR 为其合并后的统一交付 See merge request: cann/ops-math!6621	15 days ago
CMakeLists.txt	feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Co-authored-by: Dazhou_Cao<caodazhou@huawei.com> # message auto-generated for no-merge-commit merge: !6621 merge migrate-gelocal-delivery into master feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Created-by: Dazhou_Cao Commit-by: Dazhou_Cao Merged-by: cann-robot Description: ## 描述汇总迁移 gelocal（RT2.0）算子交付件： - 基础/GE 控制流算子：size、rank、LoopCond、NextIteration、RefIdentity、SwitchByIndex、aipp_data、End、squeezev2、unsqueezev2，以及 enter/exit/merge/switch/switch_n/ref_* 等 - 子图控制流算子：If、Case、For、PartitionedCall、While（原型注册 + RT2.0 推导注册 + UT + README，If/Case/For/PartitionedCall 含 GeIR 示例；均归入 ge 分类） - TF 控制流插件：If/_If/StatelessIf、While/_While/StatelessWhile、Case/StatelessCase、PartitionedCall/StatefulPartitionedCall 变体合并映射 - reshape：infershape 对齐 RT1.0 语义、allowzero/shape 属性注册、信息库 - 文档：README 产品支持表统一（全部√）、gather_shapes 归入 conversion、op_list 同步登记 ## 测试 - `bash build.sh --ophost --opgraph -j8` 构建通过（含各算子 REG_OP 注册校验） - 宿主侧全量 UT：2594/2594 PASSED（384 test suites） ## 说明 - 内容汇总自此前迁移 PR（6474 / 6522 / 6208）的交付件，本 PR 为其合并后的统一交付 See merge request: cann/ops-math!6621	15 days ago
README.md	feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Co-authored-by: Dazhou_Cao<caodazhou@huawei.com> # message auto-generated for no-merge-commit merge: !6621 merge migrate-gelocal-delivery into master feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Created-by: Dazhou_Cao Commit-by: Dazhou_Cao Merged-by: cann-robot Description: ## 描述汇总迁移 gelocal（RT2.0）算子交付件： - 基础/GE 控制流算子：size、rank、LoopCond、NextIteration、RefIdentity、SwitchByIndex、aipp_data、End、squeezev2、unsqueezev2，以及 enter/exit/merge/switch/switch_n/ref_* 等 - 子图控制流算子：If、Case、For、PartitionedCall、While（原型注册 + RT2.0 推导注册 + UT + README，If/Case/For/PartitionedCall 含 GeIR 示例；均归入 ge 分类） - TF 控制流插件：If/_If/StatelessIf、While/_While/StatelessWhile、Case/StatelessCase、PartitionedCall/StatefulPartitionedCall 变体合并映射 - reshape：infershape 对齐 RT1.0 语义、allowzero/shape 属性注册、信息库 - 文档：README 产品支持表统一（全部√）、gather_shapes 归入 conversion、op_list 同步登记 ## 测试 - `bash build.sh --ophost --opgraph -j8` 构建通过（含各算子 REG_OP 注册校验） - 宿主侧全量 UT：2594/2594 PASSED（384 test suites） ## 说明 - 内容汇总自此前迁移 PR（6474 / 6522 / 6208）的交付件，本 PR 为其合并后的统一交付 See merge request: cann/ops-math!6621	15 days ago

RefMerge

产品支持情况

产品	是否支持
Ascend 950PR&950DT系列产品	√
Atlas A3系列产品	√
Atlas A2系列产品	√
Atlas 200I/500 A2推理产品	√
Atlas推理系列产品	√
Atlas训练系列产品	√

功能说明

RefMerge为控制流基础算子，功能与Merge一致：将输入x中第一个变为可用的张量转发到输出y，并在value_index中给出该张量在输入中的索引。区别在于RefMerge用于引用（ref）类型的数据。RefMerge通常与RefSwitch算子配合实现分支汇聚。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	动态输入	输入张量，其中一个会变为可用。各输入张量应使用相同的数据类型；形状推导会合并各输入的维度信息。	float16、float32、float64、int8、int16、int32、int64、uint8、uint16、uint32、uint64、bool	ND
y	输出	可用的张量，数据类型与x一致。	float16、float32、float64、int8、int16、int32、int64、uint8、uint16、uint32、uint64、bool	ND
value_index	输出	标量，被选中的输入张量在输入列表中的索引。	DT_INT32	ND

约束说明

输入x的个数必须大于等于1，否则形状推导失败。
各输入x的数据类型必须一致；数据类型推导会逐个检查输入类型，不一致时返回失败。输出y的数据类型与输入一致，value_index为DT_INT32。
若各输入x的维度数不一致或存在未知秩输入，则输出y推导为未知秩（[-2]），不因秩不一致而返回推导失败。
对于某一维度，若各输入取值不一致或存在未知维（-1），则该维推导为-1（未知）。

调用说明

RefMerge算子为控制流基础算子，通常作为图中分支汇聚节点使用，不单独调用。

调用方式	调用样例	说明
图模式调用	test_geir_ref_merge	通过算子IR构图方式调用RefMerge算子