KthValue、Median、NanMedian 只需要目标秩对应的元素,通用完整排序和反复读取整行会带来额外开销。本次在 arch35 上按 dtype、轴长、行数、目标秩和 UB 容量细化选择算法与归并路由,减少不必要的排序、数据扫描和标量控制,保留各算子的 NaN、索引及中位数语义。
实现 PR:https://gitcode.com/cann/ops-nn/pull/10411 关联的 Math 优化 PR:https://gitcode.com/cann/ops-math/pull/5672
背景与目标
KthValue、Median、NanMedian 只需要目标秩对应的元素,通用完整排序和反复读取整行会带来额外开销。本次在 arch35 上按 dtype、轴长、行数、目标秩和 UB 容量细化选择算法与归并路由,减少不必要的排序、数据扫描和标量控制,保留各算子的 NaN、索引及中位数语义。
关键方案
验证与验收口径
实现 PR:https://gitcode.com/cann/ops-nn/pull/10411
关联的 Math 优化 PR:https://gitcode.com/cann/ops-math/pull/5672