| Files | Last commit | Last update |
|---|---|---|
StackBallQuery 增加950支持 Co-authored-by: wang-shilong32<wangshilong23@huawei.com> # message auto-generated for no-merge-commit merge: !4783 merge simt_stack_ball into master StackBallQuery 增加950支持 Created-by: wang-shilong32 Commit-by: wang-shilong32 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 增加StackBallQuery 950支持，修复kernel不支持int64 batchcount的问题，并添加对应的infershape、tiling校验和测试UT。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/2874 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 本地出包验证kernel、geir，UT测试，冒烟测试。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4783 | 2 months ago | |
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 |
| Files | Last commit | Last update |
|---|---|---|
| 2 months ago | ||
| 7 days ago |