| Files | Last commit | Last update |
|---|---|---|
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 | 5 days ago |