| fix splitd's infershape Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !6253 merge dev into master fix splitd's infershape Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述 修复SplitD infershape在处理负数split_dim时的校验顺序问题。 CheckSplitParams原先在负维度归一化前直接调用GetDim(splitDim)进行整除校验。负数索引会被转换为非法的size_t索引,使校验结果依赖无效维度哨兵值,可能错误拒绝合法输入。本次修改在维度范围校验通过后,先将负数split_dim转换为对应的非负索引,再执行整除校验,与canndev中的rt2实现保持一致。 ## 关联的Issue Fixes https://gitcode.com/cann/ops-math/issues/3506 ## 测试 执行SplitD op-host单元测试: bash bash build.sh -u --ophost --soc=ascend910b --ops=split_d -j16 测试结果:[ PASSED ] 3 tests. 覆盖场景: - 正常正维度切分。 - split_dim=-1负维度切分。 - 输入维度不能被num_split整除时返回失败。 ## 文档更新 无文档更新。 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!6253 | 5 天前 |