| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
RadixTopK关闭自动同步 Co-authored-by: huang-wei-chen<chenhuangwei1@huawei.com> # message auto-generated for no-merge-commit merge: !4184 merge radixtopk_without_autosync into master RadixTopK关闭自动同步 Created-by: huang-wei-chen Commit-by: huang-wei-chen Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 关闭RadixTopK自动同步，手动插入同步，优化算子性能。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> 关联Issue [#2286](https://gitcode.com/cann/ops-math/issues/2286) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 算子泛化200+用例精度通过，性能提升2%-5% ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [x] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4184 | 26 天前 | |
feat: 支持 kernel_src 机制，允许算子通过 CMake 指定自定义 kernel 入口文件 Co-authored-by: songkai111<songkai111@noreply.gitcode.com> # message auto-generated for no-merge-commit merge: !3141 merge revert-mr-3138-1780580007491-auto into master feat: 支持 kernel_src 机制，允许算子通过 CMake 指定自定义 kernel 入口文件 Created-by: songkai111 Commit-by: songkai111 Merged-by: cann-robot Description: ## 描述 支持 kernel_src 机制，允许算子通过 CMake 的 add_kernel_sources() 函数指定自定义 kernel 入口文件，而非默认使用算子名.cpp。同步自 ops-nn PR#3936 的 cmake/scripts 改动。 ### 改动内容 **CMake 构建系统（4个文件）**： - cmake/func.cmake：add_kernel_sources() 函数新增 KERNEL_SRC 单值参数，支持指定算子入口文件名（不含.cpp后缀），缺省使用算子名；将 KERNEL_SRC 信息收集到 KERNEL_SRC_LIST 缓存变量中 - cmake/opbuild.cmake：在 gen_aclnn_with_opdef() 构建目标中新增构建后调用 insert_kernel_src.py 脚本的步骤，分别处理 ASCEND_AUTOGEN_PATH、inner、exc 三个目录下的 ini 文件 - cmake/ut.cmake：AddOpsTestCase() 中根据 KERNEL_SRC_LIST 查找自定义 kernel 入口文件对应的源文件路径 - cmake/variables.cmake：新增 KERNEL_SRC_LIST 缓存变量，用于存储 kernel 源文件信息 **Python 脚本（4个文件）**： - scripts/util/insert_kernel_src.py：新增脚本，解析 KERNEL_SRC_LIST 并在 aic-{compute_unit}-ops-info.ini 文件中插入或更新 kernelSrc.value 字段 - scripts/util/opdesc_parser.py：OpDesc 类新增 kernel_src 属性和 parse_kernel_src() 方法，解析 ini 文件中的 kernelSrc.value 配置 - scripts/util/ascendc_impl_build.py：在 write_adapt() 和 write_impl() 中使用 kernel_src 替代 op_file 作为源文件名，未指定时回退到 op_file - scripts/util/parse_ini_to_json.py：解析 ini 时跳过 kernelSrc 键，避免写入 JSON - scripts/kernel/binary_script/gen_ops_compile_ini.py：参数校验从 5 个调整为 6 个，适配新增参数 **算子示例（2个文件）**： - math/add/op_kernel/CMakeLists.txt：为 add 算子的 ascend950 和 mc62 两个 COMPUTE_UNITS 配置 KERNEL_SRC 为 arch35/add.cpp - math/add/op_kernel/add_apt.cpp → math/add/op_kernel/arch35/add.cpp：将 add_apt.cpp 重命名为 arch35/add.cpp，更新文件内引用路径（arch35/add_dag.h → add_dag.h，arch35/add_struct.h → add_struct.h） ## 关联的Issue - #1859 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3141 | 2 个月前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 26 天前 | ||
| 2 个月前 |