已关闭
[Requirement|需求建议]: aclnnMedian 系列接口接入专用 Median/NanMedian L0 算子 #5337
黄晓彬创建于  12 天前关闭于  10 天前
黄晓彬成员
12 天前 创建

Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.

Backgroud(背景信息)

历史 Issue #3524 在 RegBase 架构下将 aclnnMedianDim 接入 KthValue,以替代 Sort + GatherV2 + Cast 组合流程。但 Median/NanMedian 不应复用 KthValue 作为实现入口,需要改为调用语义匹配的专用 Median/NanMedian L0 算子,并覆盖 aclnnMedian、aclnnMedianDim、aclnnNanMedian、aclnnNanMedianDim 四个接口。

Origin(信息来源)

自规划;对历史 Issue #3524 的接入方案进行修正。

Benefit / Necessity (价值/作用)

在 RegBase 且专用 L0 算子可用时,减少 Sort/Gather 等组合算子带来的中间内存和调度开销;在能力及性能允许的非末轴场景避免 transpose。非 RegBase 或专用 L0 符号不存在时保留原始实现,确保兼容性。

Design(设计方案)

  1. 为 Median/NanMedian 增加 weak L0 声明,使未安装 custom vendor 包时可安全回退。
  2. 仅在 RegBase 且对应 L0 符号存在时启用新路径;否则执行原 Sort/Gather 等实现。
  3. aclnnMedian/aclnnNanMedian 展平输入后直接调用对应 L0。
  4. aclnnMedianDim/aclnnNanMedianDim 对末轴直接调用;非末轴仅在轴长 2~2048 且不命中小 inner、大 outer 的性能排除条件时免 transpose,否则先将目标轴 transpose 到末轴再调用。
  5. 保持原有参数校验、输出 reshape/cast 及 fallback 行为不变。
likedislike
黄晓彬成员
12 天前 添加了label:requirement
黄晓彬成员
12 天前 修改了issue 的描述
CANN-robotCANN-robot成员
10 天前 关闭了 issue
CANN-robotCANN-robot成员
10 天前 添加了label:resolved