| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
feat(square_sum_all): add Ascend 950 support for ND, NCHW and NHWC Co-authored-by: raoliang_sac<raoliang4@huawei.com> # message auto-generated for no-merge-commit merge: !4555 merge feature/square-sum-all-ascend950 into master feat(square_sum_all): add Ascend 950 support for ND, NCHW and NHWC Created-by: raoliang_sac Commit-by: raoliang_sac Merged-by: cann-robot Description: ## 描述 - 新增 SquareSumAll 的 Ascend 950 / arch35 实现，分别计算两个同 shape FP32 张量的全元素平方和，输出两个 FP32 标量。 - 补齐 GE 图注册、InferShape/InferDataType、Host Tiling、RegBase Kernel、binary 配置及测试资产。 - Ascend 950 当前对外支持以下格式： - ND 输入对应 ND 输出，输入 rank 范围为 [0, 8]。 - NCHW、NHWC 输入必须为 rank 4，输出统一为 ND。 - x1、x2 必须同 shape、同格式，不支持广播和空 Tensor。 - Ascend 950 当前不支持 FRACTAL_Z、C1HWNCoC0、NC1HWC0。Host Tiling 会在 shape 校验和 Kernel 执行前提前拒绝。 - 私有格式对应的注册、6 条 binary 路由和 Kernel 实现资产继续保留，便于后续按版本放开，但本版本不声明支持。 - 跨核归约使用独占 workspace 和固定顺序合并，不使用原子累加。 - 本算子仅支持 GE 图模式，构建配置为 aclnn_exclude，不提供 aclnn 接口。 ## 关联的Issue #2754 ## 测试 - Host UT：36/36 PASS，其中 Tiling 26 条、InferShape 10 条；覆盖 ND/NCHW/NHWC 正向路径、三个私有格式拦截及非法格式组合。 - Ascend 950 release 包构建和部署成功，Host SO 校验一致，6 组 .o/.json binary 资产保留。 - TTK Kernel release： - NCHW、NHWC 两条公有格式用例 2/2 PASS。 - 均命中对应 binary，双输出精度 100%，内存检查通过。 - 私有格式负向板测： - NC1HWC0 binary 保持可匹配。 - Host Tiling 随后按预期返回 OPTILING_FAILURE / INVALID_TILING，Kernel 未执行。 - TTK GEIR：NCHW、NHWC 全链路 2/2 PASS，双输出精度均为 100%。 - 测试生成物一致性检查通过：Kernel 1004 条、GEIR 152 条；正向用例不包含私有格式。 - pre-commit、clang-format、ruff、codespell、OAT 和 README 链接检查全部通过。 ## 文档更新 - 更新 math/square_sum_all/README.md，仅声明 ND、NCHW、NHWC，并明确三个私有格式当前不支持、后续可能放开。 - 更新 docs/zh/op_list.md，登记 SquareSumAll。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4555 | 1 个月前 |