| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
refactor: 将arch35算子kernel中MicroAPI接口统一迁移为Reg接口 Co-authored-by: tianqiguang<tianqiguang@huawei.com> # message auto-generated for no-merge-commit merge: !4187 merge master into master refactor: 将arch35算子kernel中MicroAPI接口统一迁移为Reg接口 Created-by: tianqiguang Commit-by: tianqiguang Merged-by: cann-robot Description: ## 描述 将 math 模块下 80+ 算子的 arch35 kernel 实现从 MicroAPI 接口迁移为 Reg（寄存器）接口，统一编程范式。 ### 改动范围 涉及算子: abs, abs_grad, add_n, adjacent_difference, arg_max_with_value, assign_add, assign_sub, atan2, bitwise_xor, cast, cdist, cdist_grad, ceil, complex_abs, cummin, cumsum, div, dot, exp, eye, floor, floor_div, floor_mod, fused_mul_add_n, grouped_bias_add_grad, inv, is_close, kth_value, lerp, lin_space, log, log1p, logical_or, mod, nan_to_num, neg, pow, power, reciprocal, reduce_all, reduce_any, reduce_log_sum_exp, reduce_nansum, reduce_prod, reduce_var, right_shift, rint, round, rsqrt, rsqrt_grad, sign, signbit, sinh, sort, sort_with_index, tanh, tanh_grad, tensor_equal, top_k_v2, trunc, truncate_div 等 ### 改动方式 - 将 MicroAPI 调用替换为等价 Reg API 调用 - 适配 arch35（ascend950）指令集特性 - 保持算子功能不变 ## 测试 - 涉及算子全量回归测试 ## 类型标签 - [x] 重构 ## 关联的Issue - #2295 See merge request: cann/ops-math!4187 | 27 天前 | |
kth_value: 新增radix_select和small_axis_short_rank_select调度策略 Co-authored-by: ConanHuang<huangxiaobin1@huawei.com> # message auto-generated for no-merge-commit merge: !3804 merge master into master kth_value: 新增radix_select和small_axis_short_rank_select调度策略 Created-by: ConanHuang Commit-by: ConanHuang Merged-by: cann-robot Description: ## 描述 当前 KthValue 算子已支持多种排序/选择策略，但在部分典型场景仍存在性能优化空间： 大型 last-axis 场景下，现有 radix_more_core 路径需要完成较完整的 radix sort 流程，再提取第 k 个元素。对于 kth_value 只需要单个 kth value/index 的语义而言，全量排序存在冗余计算。 小轴 int64/uint64 场景下，若 k 靠近头部或尾部，现有 small-axis two-stage / insertion 路径仍会完成完整排序，再提取第 k 个元素。对于短 rank 场景，可以只维护少量候选元素，避免全排序开销。 本需求在现有 KthValue 多策略框架基础上新增两条专项选择路径：radix_select 和 small_axis_short_rank_select，进一步减少不必要排序计算，提升大轴和小轴短 rank 场景性能。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/2155 ## 测试 1 kthvalue 108个ST用例功能、性能达标 2 sort 331个ST用例功能、性能达标 3 二级冒烟通过 ## 文档更新 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [x] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3804 | 1 个月前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 27 天前 | ||
| 1 个月前 |