Star0

Star0

cann-robotfix(arch35): 修复 InTrainingUpdateGrad NDC1HWC0 与 MultiAddRmsNormDynamicQuant 动态输入取值；8 算子 golden 适配三方对比

44105c3e创建于 8月11日历史提交

文件	最后提交记录	最后更新时间
framework	non_zero_with_value, multilabel_margin_loss和group_norm_silu_quant下一代算子支持 Co-authored-by: tangpingchuan<tangpingchuan@huawei.com> # message auto-generated for no-merge-commit merge: !7599 merge master into master non_zero_with_value, multilabel_margin_loss和group_norm_silu_quant下一代算子支持 Created-by: zl_hw Commit-by: tangpingchuan Merged-by: cann-robot Description: ## 描述支持下一代non_zero_with_value, multilabel_margin_loss和group_norm_silu_quant算子 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/4221 ## 测试 ut/st ## 文档更新 README.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!7599	1 个月前
op_graph	NonZeroWithValue下一代芯片x/value 扩展支持全 12 种 dtype Co-authored-by: tangpingchuan<tangpingchuan@huawei.com> # message auto-generated for no-merge-commit merge: !7932 merge feat/non_zero_with_value-dtype-arch35 into master NonZeroWithValue下一代芯片x/value 扩展支持全 12 种 dtype Created-by: zl_hw Commit-by: tangpingchuan Merged-by: cann-robot Description: ## 描述 NonZeroWithValue 算子proto原型和实际实现之间支持dtype范围存在差异，下一代芯片扩展支持全 12 种 dtype，保持原型对齐 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/4320 ## 测试 ut/st ## 文档更新 README ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!7932	1 个月前
op_host	fix: 8 算子检视问题整改 + ScatterDiv 整数除零取值域资料补充 Co-authored-by: tangpingchuan<tangpingchuan@huawei.com> # message auto-generated for no-merge-commit merge: !8225 merge fix/8ops-review-issues into master fix: 8 算子检视问题整改 + ScatterDiv 整数除零取值域资料补充 Created-by: zl_hw Commit-by: tangpingchuan Merged-by: cann-robot Description: ## 描述算子检视问题清理及资料优化 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/4569 ## 测试 ut/st ## 文档更新 README.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!8225	1 个月前
op_kernel	NonZeroWithValue下一代芯片x/value 扩展支持全 12 种 dtype Co-authored-by: tangpingchuan<tangpingchuan@huawei.com> # message auto-generated for no-merge-commit merge: !7932 merge feat/non_zero_with_value-dtype-arch35 into master NonZeroWithValue下一代芯片x/value 扩展支持全 12 种 dtype Created-by: zl_hw Commit-by: tangpingchuan Merged-by: cann-robot Description: ## 描述 NonZeroWithValue 算子proto原型和实际实现之间支持dtype范围存在差异，下一代芯片扩展支持全 12 种 dtype，保持原型对齐 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/4320 ## 测试 ut/st ## 文档更新 README ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!7932	1 个月前
tests	fix(arch35): 修复 InTrainingUpdateGrad NDC1HWC0 与 MultiAddRmsNormDynamicQuant 动态输入取值；8 算子 golden 适配三方对比 Co-authored-by: tangpingchuan<tangpingchuan@huawei.com> # message auto-generated for no-merge-commit merge: !8492 merge fix/op-binary-json-missing into master fix(arch35): 修复 InTrainingUpdateGrad NDC1HWC0 与 MultiAddRmsNormDynamicQuant 动态输入取值；8 算子 golden 适配三方对比 Created-by: zl_hw Commit-by: tangpingchuan Merged-by: cann-robot Description: ## 描述按照ttk最新支持规范更新ttk golden ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/4685 ## 测试 st ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!8492	1 个月前
CMakeLists.txt	non_zero_with_value, multilabel_margin_loss和group_norm_silu_quant下一代算子支持 Co-authored-by: tangpingchuan<tangpingchuan@huawei.com> # message auto-generated for no-merge-commit merge: !7599 merge master into master non_zero_with_value, multilabel_margin_loss和group_norm_silu_quant下一代算子支持 Created-by: zl_hw Commit-by: tangpingchuan Merged-by: cann-robot Description: ## 描述支持下一代non_zero_with_value, multilabel_margin_loss和group_norm_silu_quant算子 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/4221 ## 测试 ut/st ## 文档更新 README.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!7599	1 个月前
README.md	NonZeroWithValue下一代芯片x/value 扩展支持全 12 种 dtype Co-authored-by: tangpingchuan<tangpingchuan@huawei.com> # message auto-generated for no-merge-commit merge: !7932 merge feat/non_zero_with_value-dtype-arch35 into master NonZeroWithValue下一代芯片x/value 扩展支持全 12 种 dtype Created-by: zl_hw Commit-by: tangpingchuan Merged-by: cann-robot Description: ## 描述 NonZeroWithValue 算子proto原型和实际实现之间支持dtype范围存在差异，下一代芯片扩展支持全 12 种 dtype，保持原型对齐 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/4320 ## 测试 ut/st ## 文档更新 README ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!7932	1 个月前

NonZeroWithValue

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	√
Atlas 训练系列产品	×

功能说明

算子功能：找出2D输入x的非零元素，静态max-size一次返回value、index、count三个输出，有效长度由count给出。
计算说明：
- value：非零元素的值。
- index：非零元素坐标，坐标主序展平（[2, numel]，前半段为行号、后半段为列号）。
- count：非零元素个数。
- 非零判定遵循IEEE语义：nan、±inf计为非零，-0.0视为零；按行主序遍历，结果确定。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	待计算的目标张量，严格2D。	DOUBLE, FLOAT, FLOAT16, INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, BOOL	ND
value	输出	非零元素的值，数据类型与x保持一致，静态max-size，shape为[row*col]。	DOUBLE, FLOAT, FLOAT16, INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, BOOL	ND
index	输出	非零元素坐标，坐标主序展平，shape为[2rowcol]。	INT32	ND
count	输出	非零元素个数，shape为[1]。	INT32	ND
transpose	属性	标识index排布，仅支持true（坐标主序）。	BOOL	-
dtype	属性	控制index输出数据类型，仅支持INT32。	INT	-

约束说明

输入x为严格2D，数据类型支持12类：DOUBLE、FLOAT、FLOAT16、INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、BOOL；value数据类型与x一致，index/count恒为INT32。
输入x的元素总数（row×col）不超过INT32_MAX（2³¹-1）：坐标输出index为INT32，元素总数超限时坐标无法表示，tiling校验拒绝。
输出为静态max-size，有效长度由count给出。
确定性计算：默认确定性实现。

调用说明

调用方式	样例代码	说明
图模式调用	non_zero_with_value_proto.h	通过算子IR（`REG_OP(NonZeroWithValue)`）构图方式调用NonZeroWithValue算子。