| 西工大智能感知交互实验室-AscendC实现TensorMove算子贡献 Co-authored-by: qq_61939128<942570045@qq.com> # message auto-generated for no-merge-commit merge: !3892 merge feat/tensor-move-910b into master 西工大智能感知交互实验室-AscendC实现TensorMove算子贡献 Created-by: qq_61939128 Commit-by: qq_61939128 Merged-by: cann-robot Description: ## 描述 使用 AscendC 对 TBE 实现的 TensorMove 算子进行重构,实现 AscendC 版本 TensorMove 算子对 Atlas A2 训练系列产品(Ascend 910B)的适配。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/2171?ref=&did=4149755#tid-4149755 ## 测试 已完成以下验证: - 算子编译通过,能够成功生成自定义算子包。 - 自定义算子包安装成功。 - ACLNN 样例验证通过。 - AscendOpTest 精度测试通过,输出结果与 TBE 参考实现逐元素一致。 - CANNJudge 功能验收通过。 - 本地使用 msprof 对 TensorMove TBE 与 Ascend C 实现进行了性能对比验证。 - 新增 Host / Kernel UT 测试文件,覆盖 TensorMove Tiling 与 Kernel 基础逻辑。 ## 文档更新 已更新 TensorMove 算子说明文档: - experimental/conversion/tensor_move/README.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [x] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!3892 | 3 个月前 |