ops-math/conversion/batch_to_space_nd/op_host · yuwang-/ops-math

cann-robot[ascend960dt/ascend960pr] conversion/math类9个算子新增ascend960dt/ascend960pr平台支持

Files	Last commit	Last update
arch35	format: batch_to_space_nd 等算子代码格式化 Co-authored-by: zhanw_coding<zhanwang@huawei.com> # message auto-generated for no-merge-commit merge: !3822 merge fix-fmt into master format: batch_to_space_nd 等算子代码格式化 Created-by: zhanw_coding Commit-by: zhanw_coding Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> format: batch_to_space_nd 等算子代码格式化 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/2147 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码格式化 See merge request: cann/ops-math!3822	4 months ago
config	[ascend960dt/ascend960pr] conversion/math类9个算子新增ascend960dt/ascend960pr平台支持 Co-authored-by: yangjinwen<yangjinwen2@huawei.com> # message auto-generated for no-merge-commit merge: !5778 merge dev-v121 into master [ascend960dt/ascend960pr] conversion/math类9个算子新增ascend960dt/ascend960pr平台支持 Created-by: yangjinwen Commit-by: yangjinwen Merged-by: cann-robot Description: ## 描述 conversion 类 7 个算子（batch_to_space_nd、concat、concat_d、concat_v2、depth_to_space、pack、space_to_depth）与 math 类 2 个算子（erfc、rint）新增 ascend960dt / ascend960pr 平台支持，复用 arch35 tiling 框架，主要改动： - 平台注册：9 个算子 `CMakeLists.txt` 的 `COMPUTE_UNIT`/`SUPPORT_COMPUTE_UNIT` 新增 ascend960dt、ascend960pr（`TILING_DIR` 复用 arch35），`op_host/_def.cpp` 新增对应 `this->AICore().AddConfig(...)` 注册； - 预编译 binary 接入：batch_to_space_nd、concat_d、depth_to_space、pack、space_to_depth、rint 新增 `op_host/config/ascend960dt/`、`op_host/config/ascend960pr/` 下的 `_binary.json` 与 `*_simplified_key.ini`；concat_v2 为 HOSTCPU 算子（依赖 concat），无独立 binary；erfc 的 kernel 源码（`arch35/erfc.cpp`）通过 `add_kernel_sources` 的 `COMPUTE_UNITS` 纳入新平台编译； - concat 的 arch35 tiling/kernel 适配 DAV_9201 架构（仅 9201 生效，不影响既有平台路径）： - tiling 侧：DAV_9201 平台 `ubFactorDim0` 按两倍 BLOCK（64B）对齐修正；新增 dtypeSize 提升优化（末轴字节满足 128/64/32/16B 对齐时提升 dtypeSize，减少搬运次数），并注册 bitwidth=16 系列 tilingKey（12116/12126/21116/21216/30016 等）； - kernel 侧：`__NPU_ARCH__ == 9201` 下 `DataCopyPad` 使用带 enAtomicSync 参数的接口分支。 ## 关联的Issue #3222 https://gitcode.com/cann/ops-math/issues/3222 ## 测试平台注册与预编译 binary 接入部分不涉及既有 kernel 逻辑修改；concat 为 DAV_9201 架构适配新增专属分支，既有平台路径不变；可通过 PR 评论 `compile` 触发 CI 编译流水线验证。 ## 文档更新无 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5778	27 days ago
batch_to_space_nd_def.cpp	[ascend960dt/ascend960pr] conversion/math类9个算子新增ascend960dt/ascend960pr平台支持 Co-authored-by: yangjinwen<yangjinwen2@huawei.com> # message auto-generated for no-merge-commit merge: !5778 merge dev-v121 into master [ascend960dt/ascend960pr] conversion/math类9个算子新增ascend960dt/ascend960pr平台支持 Created-by: yangjinwen Commit-by: yangjinwen Merged-by: cann-robot Description: ## 描述 conversion 类 7 个算子（batch_to_space_nd、concat、concat_d、concat_v2、depth_to_space、pack、space_to_depth）与 math 类 2 个算子（erfc、rint）新增 ascend960dt / ascend960pr 平台支持，复用 arch35 tiling 框架，主要改动： - 平台注册：9 个算子 `CMakeLists.txt` 的 `COMPUTE_UNIT`/`SUPPORT_COMPUTE_UNIT` 新增 ascend960dt、ascend960pr（`TILING_DIR` 复用 arch35），`op_host/_def.cpp` 新增对应 `this->AICore().AddConfig(...)` 注册； - 预编译 binary 接入：batch_to_space_nd、concat_d、depth_to_space、pack、space_to_depth、rint 新增 `op_host/config/ascend960dt/`、`op_host/config/ascend960pr/` 下的 `_binary.json` 与 `*_simplified_key.ini`；concat_v2 为 HOSTCPU 算子（依赖 concat），无独立 binary；erfc 的 kernel 源码（`arch35/erfc.cpp`）通过 `add_kernel_sources` 的 `COMPUTE_UNITS` 纳入新平台编译； - concat 的 arch35 tiling/kernel 适配 DAV_9201 架构（仅 9201 生效，不影响既有平台路径）： - tiling 侧：DAV_9201 平台 `ubFactorDim0` 按两倍 BLOCK（64B）对齐修正；新增 dtypeSize 提升优化（末轴字节满足 128/64/32/16B 对齐时提升 dtypeSize，减少搬运次数），并注册 bitwidth=16 系列 tilingKey（12116/12126/21116/21216/30016 等）； - kernel 侧：`__NPU_ARCH__ == 9201` 下 `DataCopyPad` 使用带 enAtomicSync 参数的接口分支。 ## 关联的Issue #3222 https://gitcode.com/cann/ops-math/issues/3222 ## 测试平台注册与预编译 binary 接入部分不涉及既有 kernel 逻辑修改；concat 为 DAV_9201 架构适配新增专属分支，既有平台路径不变；可通过 PR 评论 `compile` 触发 CI 编译流水线验证。 ## 文档更新无 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5778	27 days ago
batch_to_space_nd_infershape.cpp	[Fix] BatchToSpaceND InferShape 拒绝非正 block_shape Co-authored-by: KaranocaVe<karanocave@qq.com> # message auto-generated for no-merge-commit merge: !4998 merge fix/batch-to-space-block-shape into master [Fix] BatchToSpaceND InferShape 拒绝非正 block_shape Created-by: KaranocaVe Commit-by: KaranocaVe Merged-by: cann-robot Description: ## 修复摘要 BatchToSpaceND 的文档要求 `block_shape` 的每个元素严格大于 0，但 `conversion/batch_to_space_nd/op_host/batch_to_space_nd_infershape.cpp` 原先只在 batch 计算时拒绝 0，负值会参与除法和空间维度计算，生成负输出 shape。本修复对常量 `block_shape` 统一执行正值校验，并在 unknown-rank 输入返回前完成该校验，避免非法常量绕过。 ## 变更范围和兼容性 - 修改 `conversion/batch_to_space_nd/op_host/batch_to_space_nd_infershape.cpp`，在 shape 推导前拒绝所有 `block_shape <= 0` 的常量值。 - 修改 `conversion/batch_to_space_nd/tests/ut/op_host/test_batch_to_space_nd_infershape.cpp`，增加已知 rank、unknown rank 的负值拒绝测试，并保留零值和合法输入覆盖。 - 合法正值 block_shape、未知输入 rank 和动态 block_shape 的既有处理保持不变。 ## 测试结果 - `cmake --build <build> --target math_op_host_ut -j2`：退出码 `0`。 - `math_op_host_ut --gtest_filter='BatchToSpaceNDInfershapeTest.*' --gtest_repeat=5`： 29 个测试每轮通过，5/5 轮通过，退出码 `0`。 - 合法 `[2,2]` 返回成功并生成 `[1,2,2,1]`；`[-1,2]` 在已知 rank 和 unknown rank 两种场景均返回 `GRAPH_FAILED`；`[0,2]` 仍返回失败。 ## 关联的 Issue - https://gitcode.com/cann/ops-math/issues/2823 See merge request: cann/ops-math!4998	1 month ago