| fix: 修复 AdaCast、StackBallQuery、ProdForceSeA 算子功能与边界问题 Co-authored-by: wyy<2137553008@qq.com> # message auto-generated for no-merge-commit merge: !6726 merge master into master fix: 修复 AdaCast、StackBallQuery、ProdForceSeA 算子功能与边界问题 Created-by: wangyuyan_44 Commit-by: wyy Merged-by: cann-robot Description: ## 描述 修复 AdaCast、StackBallQuery、ProdForceSeA 的精度、输入校验与边界处理问题,并补充 ascend350 平台配置。 ### AdaCast - fp32→fp16 改用 CAST_RINT,与参考的就近取偶舍入语义对齐。 - 多核切分使用 CeilDiv 避免极大维度下的整数溢出,补充对应边界 UT;同步参数与 GE-IR 调用样例说明。 ### StackBallQuery - 补充超大 sample_num 与空 center_xyz 的边界保护,修正尾核 GM 长度和非 32B 对齐搬运,避免越界与除零。 - 修正首个命中索引、无候选点时的 -1 填充及 fp16 半径平方的舍入方式。 - 删除未使用的 fp16 计算分支、对齐缓冲区和局部变量。 ### ProdForceSeA - 输入输出 dtype 统一为 float32,补充维度、整数范围及 natoms 值一致性校验。 - Host 统一确定 nall,交叉校验输出 shape 与可用的 natoms[1];值未预取时使用输出 shape,kernel 使用已校验的 tiling 值。 - 中心力与邻居力先在寄存器内分组进行 Neumaier 补偿累加,再与 GM 中的结果及补偿项合并,减少内层循环的 GM 访问;同步 workspace 配置。 ## 关联的Issue - AdaCast:https://gitcode.com/wangyueting/changwei_liuyi/issues/36 - StackBallQuery:https://gitcode.com/wangyueting/changwei_xuejinghui/issues/8 ## 测试 - 新增 AdaCast 极大维度 tiling 边界 UT。 - PR 当前 CI 门禁已通过。 ## 类型标签 - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!6726 | 7 days ago |