| Files | Last commit | Last update |
|---|---|---|
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 | |
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 | |
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 | |
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 | |
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 | |
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 |
Switch
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR&950DT系列产品 | √ |
| Atlas A3系列产品 | √ |
| Atlas A2系列产品 | √ |
| Atlas 200I/500 A2推理产品 | √ |
| Atlas推理系列产品 | √ |
| Atlas训练系列产品 | √ |
功能说明
Switch为控制流基础算子，根据布尔标量输入pred将输入张量data转发到对应的输出端口：当pred为true时，data转发到output_true；否则转发到output_false。该算子的两个输出output_true、output_false的形状与数据类型均与输入data一致，通常与Merge算子配合实现分支控制。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| data | 输入 | 待转发的张量。 | TensorType::ALL() | ND |
| pred | 输入 | 布尔标量，决定数据被转发到哪个输出端口。 | DT_BOOL | ND |
| output_false | 输出 | 当pred为false时，data转发到该输出，形状与数据类型同data。 | TensorType::ALL() | ND |
| output_true | 输出 | 当pred为true时，data转发到该输出，形状与数据类型同data。 | TensorType::ALL() | ND |
约束说明
- 输入
pred必须为布尔标量（rank为0或未知rank），否则形状推导失败。 - 输入
pred的数据类型必须为DT_BOOL，否则数据类型推导失败。
调用说明
Switch算子为控制流基础算子，通常作为图中分支选择节点使用，不单独调用。
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| 图模式调用 | test_geir_switch | 通过算子IR构图方式调用Switch算子 |