| 修复topkv2算子的oom问题 Co-authored-by: caoyan_huawei<caoyan11@huawei.com> # message auto-generated for no-merge-commit merge: !3960 merge cy_br_07_09_oom into master 修复topkv2算子的oom问题 Created-by: caoyan_huawei Commit-by: caoyan_huawei Merged-by: cann-robot Description: ## 描述 修复topkv2算子在处理超大尾轴时的oom问题。 【问题原因】kernel 实际是两段独立使用，独立对齐32字节，在tiling侧实际申请时是sortOutValueGM_和outValueDbWK_先乘以2，然后在结果的基础上对齐32字节，导致实际使用的空间要比申请的大32字节，导致sortOutValueGM_这块GM在写入时会超过预配的空间； 【修复方案】在tiling侧预配GM空间时先对齐32字节，然后再乘以2，和kernel保持一致； ## 关联的Issue https://gitcode.com/cann/ops-math/issues/2203 ## 测试 ST，冒烟测试通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3960 | 1 个月前 |
