ops-math_work/common/inc/op_graph · Nice_try/ops-math_work

Files	Last commit	Last update
fusion_pass	fix patterns compatibility for random fusionpass Co-authored-by: biabu111<hebaojing1@huawei.com> # message auto-generated for no-merge-commit merge: !3907 merge master into master fix patterns compatibility for random fusionpass Created-by: biabu111 Commit-by: biabu111 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> DropOutV3FusionPass,DropOutV3SplitFusionPass,RandomStandardNormalFusionPass,RandomUniformFusionPass,BernoulliFusionPass,TruncatedNormalFusionPass 兼容性处理 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/2175 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 冒烟已过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3907	3 months ago
graph_infer_comm_funcs.h	fix: input_util.h 等添加头文件保护 Co-authored-by: zhanw_coding<zhanwang@huawei.com> # message auto-generated for no-merge-commit merge: !2722 merge fix-header into master fix: input_util.h 等添加头文件保护 Created-by: zhanw_coding Commit-by: zhanw_coding Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> fix: input_util.h 等公共头文件添加头文件保护 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/1547 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!2722	5 months ago
op_math_proto_extend.h	feat: ConfusionMatrix 算子下一代实现 Co-authored-by: SeanDictionary<b24081019@njupt.edu.cn> # message auto-generated for no-merge-commit merge: !5444 merge feat/confusion-matrix into master feat: ConfusionMatrix 算子下一代实现 Created-by: SeanDictionary Commit-by: SeanDictionary Merged-by: cann-robot Description: ## 描述新增 ConfusionMatrix 算子 Ascend950（arch35）实现。 - 功能：对每个索引 i，将 weights[i]（缺省为 1）累加到输出矩阵 y[labels[i], predictions[i]]，输出 num_classes×num_classes 混淆矩阵，兼容 TF `tf.math.confusion_matrix`。 - kernel 侧：SIMT 范式（参照 bincount 骨架），2D scatter-add；按输出矩阵与 UB 容量关系分四种调度策略（full_load / batch_load / not_full_load / determine），determine 模式按输出行切核保证确定性，int8/uint8 输出固定走 determine（窄整型不走原子加）。 - host 侧：arch35 tiling（策略选择 + 结构化宏校验 shape/dtype/num_classes）；op_graph 双挂 InferShape + InferDataType（输出 dtype 由属性 dtype 字符串白名单映射）。 - dtype 组合：labels/predictions（dtype 相同）× weights/output（dtype 相同）共 25 种组合，输出 dtype 由属性 dtype 决定；1 ≤ num_classes ≤ 4096。 - 原型去重：`common/inc/op_graph/op_math_proto_extend.h` 中移除 ConfusionMatrix 的 REG_OP 声明——本算子目录 `op_graph/confusion_matrix_proto.h` 已提供带 `OPS_PROTO_DEF_CONFUSIONMATRIX` 去重宏保护的版本，避免同 TU 重复注册。 - tf-plugin：由 `common/src/framework/confusion_matrix_tf_plugin.cpp` 统一提供（#5653 已合入），本 PR 不重复交付。 ## 关联的Issue [#3211](https://gitcode.com/cann/ops-math/issues/3211) ## 测试 - 泛化：kernel 模式 1000 例（25 dtype 组合 × num_classes 1~4096 探顶 × N 0~100000 × 4 调度分支），REAL_FAIL=0（995 PASS + 5 例 fp16 深累加顺序舍入差异，与 built-in TBE 同数值行为）。 - 通路：GEIR 55 例全连通，浮点输出 23/23 三方比对（NPU / CPU golden / GPU）PASS；tf 通路经 pb 解析验证。 - 精度：三方 12/12 PASS（含 int8/uint8 饱和回绕、inf/nan 传播，NPU-GPU bitwise 一致）。 - 性能：150 例对标 A100（torch scatter_add 拼接基准），NPU 中位快约 2.6×，按 shape 分布如实呈现。 - UT：tiling（arch35）+ infershape 单测随包；ST：`tests/st/arch35/ttk_kernel_confusion_matrix_st.csv`。 ## 文档更新 - 新增 `math/confusion_matrix/README.md`（产品支持情况/功能说明/参数说明/约束说明/调用说明）。 - `docs/zh/op_list.md` 登记 confusion_matrix 条目。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5444	1 month ago