master

cann-robotfix: 修复 Median 和 NanMedian 负 subnormal 的 NaN 计数

6d3a3af6Add on 21 days agoHistory

Files	Last commit	Last update
arch35	fix: 修复 Median 和 NanMedian 负 subnormal 的 NaN 计数 Co-authored-by: ConanHuang<huangxiaobin1@huawei.com> # message auto-generated for no-merge-commit merge: !12490 merge fix/median-subnormal-nan-count into master fix: 修复 Median 和 NanMedian 负 subnormal 的 NaN 计数 Created-by: ConanHuang Commit-by: ConanHuang Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 修复 Ascend 950 Median / NanMedian 在多核 radix 路径中将负 subnormal 误计为 NaN，导致中位数及索引错误的问题。输入本身无需包含 NaN，正确中位数也不一定是 subnormal。 PreProcess 会原地将输入转换为 radix key。原先随后调用的 OnRadixInputLoaded 读取已转换的数据，例如 FP32 0x80000001 对应 key 0x7ffffffe，按浮点解释即为 NaN，影响 Median 的 NaN 传播和 NanMedian 的中位秩计算。 本次将 OnRadixInputLoaded 移到 PreProcess 之前，针对原始输入统计 NaN，并说明调用顺序的原因。复用已有计算、同步与 buffer；KthValue / Unique 的该 hook 为空操作。Median 和 NanMedian 的库内 ST CSV 各增加一条 FP32 负 subnormal 零容差回归用例，shape 为 (1,1,151992)、dim=-1。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/7143 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 本机 Ascend950DT_9574，TTK v3.0.0（8b3f093），release kernel 直调，显式加载各算子 golden 插件。 | 验证范围 | 结果 | |---|---| | Median 全量 | 417/417 通过 | | NanMedian 全量 | 427/427 通过 | | 上述全量中的新增 subnormal | 各 152 条，共 304 条，均通过；覆盖 FP32/FP16/BF16 的适用浮点模板 schId 0–11 | | KthValue 共享代码回归 | 366/366 通过，含 15 条新增 subnormal | | UniqueWithCountsAndSorting 共享代码回归 | 有效参考下 254/254 通过，含 12 条新增 subnormal；values-only 通路，校验 values 和动态 shape | ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!12490	21 days ago
CMakeLists.txt	kth_value,median,nan_median算子新增ascend350平台支持 Co-authored-by: jerry_gd<jerry.luo@huawei.com> # message auto-generated for no-merge-commit merge: !10327 merge master into master kth_value,median,nan_median算子新增ascend350平台支持 Created-by: jerry_gd Commit-by: jerry_gd Merged-by: cann-robot Description: ## 描述 kth_value,median,nan_median算子新增ascend350平台支持 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5838 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> bash build.sh --pkg --soc=ascend350 --ops="kth_value,median,nan_median"，编译验证通过。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> NA ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!10327	1 month ago