| 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 | 17 days ago | |
add hard_max fusion pass Co-authored-by: wangxun21<wangxun21@huawei.com> # message auto-generated for no-merge-commit merge: !6412 merge feat/hardmax-fusion-pass into master add hard_max fusion pass Created-by: wangxun21 Commit-by: wangxun21 Merged-by: cann-robot Description: ## 描述 将 canndev 旧框架 HardMaxPass 迁移到 ops-math 新版 Pattern Fusion Pass + ES Graph Builder 框架(注册名保持 HardMaxPass,Stage kAfterInferShape)。 - 平台分支对齐旧版:950/regbase 用 ArgMaxV2+OneHot,非 regbase(910B 等)用 ArgMaxD+OneHotD - on/off 常量按输入 dtype 匹配(FP16 用半精度位模式构造),保证 OneHot/OneHotD 输出 dtype 与 HardMax 一致 - axis 越界在 MeetRequirements 拒绝融合(旧版为未定义行为,实测导致 ATC 崩溃 exit 134) - CMake DEPENDENCIES 声明 arg_max_v2/one_hot,支持 --ops=hard_max 过滤构建 - UT 6 用例覆盖 950/910B 两分支、FP32/FP16、正负 axis、非法 dtype/axis ## 验证 - 专项 UT(--ops=hard_max)6/6,全量 graph UT 136/136 - ATC(Ascend950PR_9599)OPP overlay 取证:RunPatternFusion:GraphFusionPass[HardMaxPass] matched_times=1, effected_times=1 - NPU E2E 4 用例(FP32 12x4x3 axis=1 / FP16 / 5x8x6 axis=-1 / 3x9x2x7 axis=2)均 max_abs_error=0.0 - 910B 分支(ArgMaxD+OneHotD)ATC/NPU E2E 已在cannlab单独验证 See merge request: cann/ops-math!6412 | 17 days ago |