Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
历史 Issue #3524 在 RegBase 架构下将 aclnnMedianDim 接入 KthValue,以替代 Sort + GatherV2 + Cast 组合流程。但 Median/NanMedian 不应复用 KthValue 作为实现入口,需要改为调用语义匹配的专用 Median/NanMedian L0 算子,并覆盖 aclnnMedian、aclnnMedianDim、aclnnNanMedian、aclnnNanMedianDim 四个接口。
自规划;对历史 Issue #3524 的接入方案进行修正。
在 RegBase 且专用 L0 算子可用时,减少 Sort/Gather 等组合算子带来的中间内存和调度开销;在能力及性能允许的非末轴场景避免 transpose。非 RegBase 或专用 L0 符号不存在时保留原始实现,确保兼容性。
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(设计方案)