master

cann-robotfix(space_to_batch): 补充 BF16 支持与入参/溢出校验，修正 golden 与 README 批次映射

dfe9a4c9Add on 2 days agoHistory

Files	Last commit	Last update
arch35	fix(space_to_batch): 补充 BF16 支持与入参/溢出校验，修正 golden 与 README 批次映射 Co-authored-by: guyunlong1<guyunlong1@huawei.com> # message auto-generated for no-merge-commit merge: !7096 merge fix/issue-4456-space-to-batch-hardening-20261104T150000Z-4456 into master fix(space_to_batch): 补充 BF16 支持与入参/溢出校验，修正 golden 与 README 批次映射 Created-by: guyunlong1 Commit-by: guyunlong1 Merged-by: cann-robot Description: ## 描述 针对 Issue #4456 对 SpaceToBatch 进行质量加固： - 新增 BFLOAT16 数据类型支持（op_host/space_to_batch_def.cpp、op_graph/space_to_batch_proto.h、README.md）； - Tiling：校验输入 x 各维度必须为正；H_padded/W_padded/输出 N 改为带溢出检查的加法/乘法；补 perCoreCount 零值保护； - InferShape：校验 paddings 元素个数为 [2,2]；校验维度为正及 padding 后可被 block_size 整除；增加溢出保护； - Kernel：索引与计数 uint32→uint64，避免大 shape 截断；perCoreCount 为 0 时提前返回； - README：修正输出批次映射公式（n=n_out%N、rem=n_out/N、bh=rem/bs、bw=rem%bs）； - GEIR 样例：文件改二进制读写，补充错误处理与资源释放； - golden：TensorFlow 优先 + Torch fallback，补充 __spec__/third_party/tolerance； - UT：新增零 N/H/C 维失败用例。 ## 关联的Issue Closes #4456 https://gitcode.com/cann/ops-math/issues/4456 ## 测试 - clang-format --style=file --dry-run --Werror（仓库 .clang-format，clang-format 18.1.8）：通过 - python3 -m py_compile conversion/space_to_batch/tests/assets/golden.py：通过 - 提交前 pre-commit（trailing-whitespace、end-of-file-fixer、clang-format、ruff、codespell、OAT 等）：全部通过 - 未在隔离 worktree 重新构建 UT（当前 CMake 配置 ENABLE_TEST=OFF），由 CI 编译验证 ## 文档更新 更新 conversion/space_to_batch/README.md 与 op_graph/space_to_batch_proto.h 注释。 ## 类型标签 - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!7096	2 days ago
space_to_batch_apt.cpp	feat: add space_to_batch for ascend950 Co-authored-by: wangrui<wangrui124@huawei.com> # message auto-generated for no-merge-commit merge: !3132 merge br_space2batch into master feat: add space_to_batch for ascend950 Created-by: wangrui_ Commit-by: wangrui Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 1. 新增space_to_batch ascend950实现 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/1899 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3132	5 months ago