| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
feat: migrate case_condition to conversion Co-authored-by: zhaowenrui666<zhaowenrui7@huawei.com> # message auto-generated for no-merge-commit merge: !3703 merge migrate-case-condition-conversion into master feat: migrate case_condition to conversion Created-by: zhaowenrui666 Commit-by: zhaowenrui666 Merged-by: cann-robot Description: ## Summary 迁移 CaseCondition 到 ops-math/conversion,按标准 AICPU 交付补齐: - AICPU kernel 与 case_condition_aicpu_def.cpp - op_graph / op_host - UT - README - GE IR example ## Validation - bash ./build.sh --run_example case_condition graph --soc=ascend910_93 --noexec - example 编译通过,生成 test_geir_case_condition 可执行文件 - git diff --cached --check passed before commit See merge request: cann/ops-math!3703 | 2 个月前 | |
math仓算子添加隔离宏 Co-authored-by: ayuemoon<15802951036@163.com> # message auto-generated for no-merge-commit merge: !5280 merge proto_reg_op_modify into master math仓算子添加隔离宏 Created-by: ayuemoon Commit-by: ayuemoon Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 为 CalcBucketsLimitAndOffset、CaseCondition、ConjugateTranspose、CompareAndBitpack、Conj、GetDynamicDims、InplaceTopKDistance、ProdEnvMatACalcRij、TopKPQDistance、TopKPQDistanceV2、TopKV2D、Zeta 算子添加隔离宏,防止原型重复定义。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> 无 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> 无 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!5280 | 16 天前 | |
Aicpu op [CaseCondition、Less、Log、NotEqual、ReduceAll、Rsqrt、Sqrt] open const folding Co-authored-by: FengHaozhan<fenghaozhan@huawei.com> # message auto-generated for no-merge-commit merge: !5103 merge ops-2026-08 into master Aicpu op [CaseCondition、Less、Log、NotEqual、ReduceAll、Rsqrt、Sqrt] open const folding Created-by: FengHaozhan Commit-by: FengHaozhan Merged-by: cann-robot Description: ## 描述 为 CaseCondition、Less、Log、NotEqual、ReduceAll、Rsqrt、Sqrt 等 AICPU 算子补齐 host 侧常量折叠接线: - 在目标算子 CMake 配置中开启 HOSTCPU TRUE,使 host 常量折叠目标参与构建。 - 将 AICPU kernel 注册切换为 OPS_MATH_REGISTER_CPU_KERNELV2,保留设备侧既有计算语义并满足 host 侧注册要求。 - 为 Rsqrt 和 Sqrt 的 AICPU JSON 配置增加 OPS_FLAG_OPEN。 - 为 CaseCondition 新增 RT2 InferShape(标量输出)和 InferDataType(INT32 输出)实现,并补充对应 op_host/op_graph UT;同步调整 op_graph UT 链接顺序以保证注册符号可解析。 ## 关联的Issue #2906 ## 测试 - 新增 CaseCondition RT2 InferShape UT,覆盖输出为标量的推导结果。 - 新增 CaseCondition InferDataType UT,覆盖输出类型为 INT32 的推导结果。 - 本次未在 PR 中记录实际构建、UT 执行或设备验证结果,需由 CI/验证环境补充确认 host folding SO 产物及常量折叠触发情况。 ## 文档更新 无文档变更。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:构建与注册配置调整 See merge request: cann/ops-math!5103 | 20 天前 | |
Aicpu op [CaseCondition、Less、Log、NotEqual、ReduceAll、Rsqrt、Sqrt] open const folding Co-authored-by: FengHaozhan<fenghaozhan@huawei.com> # message auto-generated for no-merge-commit merge: !5103 merge ops-2026-08 into master Aicpu op [CaseCondition、Less、Log、NotEqual、ReduceAll、Rsqrt、Sqrt] open const folding Created-by: FengHaozhan Commit-by: FengHaozhan Merged-by: cann-robot Description: ## 描述 为 CaseCondition、Less、Log、NotEqual、ReduceAll、Rsqrt、Sqrt 等 AICPU 算子补齐 host 侧常量折叠接线: - 在目标算子 CMake 配置中开启 HOSTCPU TRUE,使 host 常量折叠目标参与构建。 - 将 AICPU kernel 注册切换为 OPS_MATH_REGISTER_CPU_KERNELV2,保留设备侧既有计算语义并满足 host 侧注册要求。 - 为 Rsqrt 和 Sqrt 的 AICPU JSON 配置增加 OPS_FLAG_OPEN。 - 为 CaseCondition 新增 RT2 InferShape(标量输出)和 InferDataType(INT32 输出)实现,并补充对应 op_host/op_graph UT;同步调整 op_graph UT 链接顺序以保证注册符号可解析。 ## 关联的Issue #2906 ## 测试 - 新增 CaseCondition RT2 InferShape UT,覆盖输出为标量的推导结果。 - 新增 CaseCondition InferDataType UT,覆盖输出类型为 INT32 的推导结果。 - 本次未在 PR 中记录实际构建、UT 执行或设备验证结果,需由 CI/验证环境补充确认 host folding SO 产物及常量折叠触发情况。 ## 文档更新 无文档变更。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:构建与注册配置调整 See merge request: cann/ops-math!5103 | 20 天前 | |
Aicpu op [CaseCondition、Less、Log、NotEqual、ReduceAll、Rsqrt、Sqrt] open const folding Co-authored-by: FengHaozhan<fenghaozhan@huawei.com> # message auto-generated for no-merge-commit merge: !5103 merge ops-2026-08 into master Aicpu op [CaseCondition、Less、Log、NotEqual、ReduceAll、Rsqrt、Sqrt] open const folding Created-by: FengHaozhan Commit-by: FengHaozhan Merged-by: cann-robot Description: ## 描述 为 CaseCondition、Less、Log、NotEqual、ReduceAll、Rsqrt、Sqrt 等 AICPU 算子补齐 host 侧常量折叠接线: - 在目标算子 CMake 配置中开启 HOSTCPU TRUE,使 host 常量折叠目标参与构建。 - 将 AICPU kernel 注册切换为 OPS_MATH_REGISTER_CPU_KERNELV2,保留设备侧既有计算语义并满足 host 侧注册要求。 - 为 Rsqrt 和 Sqrt 的 AICPU JSON 配置增加 OPS_FLAG_OPEN。 - 为 CaseCondition 新增 RT2 InferShape(标量输出)和 InferDataType(INT32 输出)实现,并补充对应 op_host/op_graph UT;同步调整 op_graph UT 链接顺序以保证注册符号可解析。 ## 关联的Issue #2906 ## 测试 - 新增 CaseCondition RT2 InferShape UT,覆盖输出为标量的推导结果。 - 新增 CaseCondition InferDataType UT,覆盖输出类型为 INT32 的推导结果。 - 本次未在 PR 中记录实际构建、UT 执行或设备验证结果,需由 CI/验证环境补充确认 host folding SO 产物及常量折叠触发情况。 ## 文档更新 无文档变更。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:构建与注册配置调整 See merge request: cann/ops-math!5103 | 20 天前 | |
Aicpu op [CaseCondition、Less、Log、NotEqual、ReduceAll、Rsqrt、Sqrt] open const folding Co-authored-by: FengHaozhan<fenghaozhan@huawei.com> # message auto-generated for no-merge-commit merge: !5103 merge ops-2026-08 into master Aicpu op [CaseCondition、Less、Log、NotEqual、ReduceAll、Rsqrt、Sqrt] open const folding Created-by: FengHaozhan Commit-by: FengHaozhan Merged-by: cann-robot Description: ## 描述 为 CaseCondition、Less、Log、NotEqual、ReduceAll、Rsqrt、Sqrt 等 AICPU 算子补齐 host 侧常量折叠接线: - 在目标算子 CMake 配置中开启 HOSTCPU TRUE,使 host 常量折叠目标参与构建。 - 将 AICPU kernel 注册切换为 OPS_MATH_REGISTER_CPU_KERNELV2,保留设备侧既有计算语义并满足 host 侧注册要求。 - 为 Rsqrt 和 Sqrt 的 AICPU JSON 配置增加 OPS_FLAG_OPEN。 - 为 CaseCondition 新增 RT2 InferShape(标量输出)和 InferDataType(INT32 输出)实现,并补充对应 op_host/op_graph UT;同步调整 op_graph UT 链接顺序以保证注册符号可解析。 ## 关联的Issue #2906 ## 测试 - 新增 CaseCondition RT2 InferShape UT,覆盖输出为标量的推导结果。 - 新增 CaseCondition InferDataType UT,覆盖输出类型为 INT32 的推导结果。 - 本次未在 PR 中记录实际构建、UT 执行或设备验证结果,需由 CI/验证环境补充确认 host folding SO 产物及常量折叠触发情况。 ## 文档更新 无文档变更。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:构建与注册配置调整 See merge request: cann/ops-math!5103 | 20 天前 | |
feat: migrate case_condition to conversion Co-authored-by: zhaowenrui666<zhaowenrui7@huawei.com> # message auto-generated for no-merge-commit merge: !3703 merge migrate-case-condition-conversion into master feat: migrate case_condition to conversion Created-by: zhaowenrui666 Commit-by: zhaowenrui666 Merged-by: cann-robot Description: ## Summary 迁移 CaseCondition 到 ops-math/conversion,按标准 AICPU 交付补齐: - AICPU kernel 与 case_condition_aicpu_def.cpp - op_graph / op_host - UT - README - GE IR example ## Validation - bash ./build.sh --run_example case_condition graph --soc=ascend910_93 --noexec - example 编译通过,生成 test_geir_case_condition 可执行文件 - git diff --cached --check passed before commit See merge request: cann/ops-math!3703 | 2 个月前 |
CaseCondition
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
| Atlas 200I/500 A2 推理产品 | × |
| Atlas 推理系列产品 | √ |
| Atlas 训练系列产品 | √ |
功能说明
- 算子功能:根据三个输入值与阈值的关系,输出离散case编号。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
x |
输入 | 长度为3的一维张量,依次表示i、j、k。 |
INT32、INT64、UINT64 | ND |
algorithm |
属性 | 当前仅支持LU。 |
STRING | - |
y |
输出 | 输出case编号。 | INT32 | ND |
约束说明
x必须为shape[3]的向量。algorithm仅支持LU。
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| 图模式调用 | test_geir_case_condition | 通过算子IR构图方式调用CaseCondition算子。 |