master

cann-robotfeat: 新增 inplace_partial_rotary_mul 算子

36984c3e创建于 6月15日历史提交

文件	最后提交记录	最后更新时间
CMakeLists.txt	feat: 新增 inplace_partial_rotary_mul 算子 Co-authored-by: wangxun21<wangxun21@huawei.com> Co-authored-by: zhu-lei0614<zhulei113@huawei.com> # message auto-generated for no-merge-commit merge: !6085 merge feat/inplace_partial_rotary_mul into master feat: 新增 inplace_partial_rotary_mul 算子 Created-by: wangxun21 Commit-by: wangxun21;zhu-lei0614 Merged-by: cann-robot Description: ## 描述 新增 posembedding/inplace_partial_rotary_mul 算子，支持原地部分旋转乘法操作。 该算子用于 RoPE（旋转位置编码）场景，支持 ascend910b、ascend910_93、ascend950 三个平台，包含 regbase 和 membase 两种 kernel 实现方式。 共新增 50 个文件，11257 行代码，包括： - op_host: tiling 定义、infershape、op_api 接口 - op_kernel: regbase (arch35) 和 membase kernel 实现 - op_graph: proto 定义 - docs: aclnn 接口文档 - examples: 测试示例代码 ## 关联的Issue #2752 ## 测试 - 已提供 examples 测试示例代码 - 需编译验证和精度测试 ## 文档更新 - 新增 aclnnInplacePartialRotaryMul.md 接口文档 - 新增 README.md 算子说明文档 ## 类型标签 - [x] ✨ 新特性 See merge request: cann/ops-transformer!6085	1 个月前
inplace_partial_rotary_mul_proto.h	feat: 新增 inplace_partial_rotary_mul 算子 Co-authored-by: wangxun21<wangxun21@huawei.com> Co-authored-by: zhu-lei0614<zhulei113@huawei.com> # message auto-generated for no-merge-commit merge: !6085 merge feat/inplace_partial_rotary_mul into master feat: 新增 inplace_partial_rotary_mul 算子 Created-by: wangxun21 Commit-by: wangxun21;zhu-lei0614 Merged-by: cann-robot Description: ## 描述 新增 posembedding/inplace_partial_rotary_mul 算子，支持原地部分旋转乘法操作。 该算子用于 RoPE（旋转位置编码）场景，支持 ascend910b、ascend910_93、ascend950 三个平台，包含 regbase 和 membase 两种 kernel 实现方式。 共新增 50 个文件，11257 行代码，包括： - op_host: tiling 定义、infershape、op_api 接口 - op_kernel: regbase (arch35) 和 membase kernel 实现 - op_graph: proto 定义 - docs: aclnn 接口文档 - examples: 测试示例代码 ## 关联的Issue #2752 ## 测试 - 已提供 examples 测试示例代码 - 需编译验证和精度测试 ## 文档更新 - 新增 aclnnInplacePartialRotaryMul.md 接口文档 - 新增 README.md 算子说明文档 ## 类型标签 - [x] ✨ 新特性 See merge request: cann/ops-transformer!6085	1 个月前