| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
topk优化特定场景性能 Co-authored-by: JimmyLam2<linjingmin2@huawei.com> # message auto-generated for no-merge-commit merge: !1864 merge master into master topk优化特定场景性能 Created-by: JimmyLam2 Commit-by: JimmyLam2 Merged-by: cann-robot Description: ## 描述 topk 在int64_t小规模shape下ai cpu相比ai core性能更快 single block场景下AscendC::Topk api在浮点场景下比AscendC::Sort api更快 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/1066 ## 测试 shape： [7478, 1, 1, 1, 9] 取top8 int64_t类型 走ai core  更改ai cpu后  ## 文档更新 NA ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [x ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!1864 | 5 个月前 | |
优化TopK算子编译构建时间 Co-authored-by: BruceLP<lipeng192@huawei.com> # message auto-generated for no-merge-commit merge: !1509 merge topk_compile into master 优化TopK算子编译构建时间 Created-by: BruceLP Commit-by: BruceLP Merged-by: cann-robot Description: ## 描述 TopK单算子编译时间超过500秒，相较于其他算子，编译构建时间较长。 优化方法: 通过宏对dtype进行判断，单次只编译一个dtype类型的代码，隔离其他类型相关的代码，从而减少单次编译构建时间 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [issues_905](https://gitcode.com/cann/ops-math/issues/905) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 同环境下测试，TopK单算子编译时间减少至200秒以内 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [x] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!1509 | 6 个月前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 5 个月前 | ||
| 6 个月前 |