已合并
optimize repeat_interleave kernel 102 tailRepeat Copy Logic #2749
陈展熹创建于 3月16日
optimize repeat_interleave kernel 102 tailRepeat Copy Logic #2749
已合并
陈展熹创建于 3月16日
陈展熹成员
3月16日

描述

优化repeat_interleave算子tailRepeat的拷贝逻辑

关联的Issue

关联Issue #1439

测试

st泛化测试、evb上板测试

文档更新

类型标签

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 陈展熹 的贡献)
陈展熹成员
3月16日 创建了 pull request,commit 2c2bfb71
CANN-robot
CANN-robot成员
3月16日 评论:

Thanks for your pull-request.
The full list of commands accepted by me can be found at here
You can get sig-info at here


PR Approval Progress

Congratulations! All modules have met the lgtm and approve requirements.

Module Approval Details

module lgtm status approve status
index chenqi317, fanqirui (2/2) chenqi317 (1/1)

💡 Tip:

  • Committer can comment /approve or /lgtm
  • Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

yourealize, thanks for your pull request. All authors of the commits have signed the CLA. 👍

likedislike
CANN-robotCANN-robot成员
3月16日 添加了label:cann-cla/yes
CANN-robotCANN-robot成员
3月16日 将FelixTang7,crystalhu,fanqirui,zhou-qilong,chenqi317,liubo75,tangweiwei2,wangyongguang,zhang-wu设为评审人
CANN-robotCANN-robot成员
3月16日 将FelixTang7,crystalhu,zhou-qilong,chenqi317,liubo75,tangweiwei2,wangyongguang,zhang-wu设为审查人
CANN-robot
CANN-robot成员
3月16日 评论:

🔵 source code change are detected, tasks labels is removed in this pull request!

likedislike
CANN-robot
CANN-robot成员
3月16日 评论:

问题/功能描述

本次PR主要优化了repeat_interleave算子内核的性能和资源利用率。修复了在处理数据块(特别是尾部数据)时,因输出缓冲区管理不当和存在冗余数据复制操作而导致的效率问题,使内核执行逻辑更加高效。

修改方案描述

修改聚焦于CopyXToMatchOutProcessCpMatchToUb两个核心函数。核心调整包括:1) 将结果复制到全局内存的操作(CopyMatchOutToY)从条件判断中移出,确保每个数据块处理后都能及时输出;2) 重构输出张量的分配逻辑,仅在需要处理新数据块时才重新分配,避免了不必要的开销;3) 移除ProcessCpMatchToUb函数循环内的冗余输出调用,统一在CopyXToMatchOut函数内处理,使数据流更清晰、高效。

likedislike
陈展熹成员
3月16日 评论:

compile

likedislike
CANN-robotCANN-robot成员
3月16日 添加了label:ci-pipeline-running
CANN-robot
CANN-robot成员
3月16日 评论:

流水线任务触发成功
任务链接 [037f7a2dbba6432d8b1c5f5e7ac4b380][流水线指导]

任务名称状态日志下载链接
解析CI分支 ✅ SUCCESS
codecheck ✅ SUCCESS >>>>>
anti_virus ✅ SUCCESS >>>>>
Check_Pr ✅ SUCCESS >>>>>
Compile_Ascend_X86_ubuntu24 ✅ SUCCESS >>>>>
Compile_Ascend_X86_mobile_station ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_single ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_experimental ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_950 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_950 ✅ SUCCESS >>>>> >>>>>
API_Check ✅ SUCCESS >>>>>
Pre_Compile ✅ SUCCESS
pre_comment ✅ SUCCESS >>>>>
Compile_Ascend_X86 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_mobile_station_ubuntu24 ✅ SUCCESS >>>>>
Compile_Ascend_X86_950_ubuntu24 ✅ SUCCESS >>>>>
StaticCheck_codespell_check ✅ SUCCESS
StaticCheck_link_validity_check ✅ SUCCESS
StaticCheck_resource_existence_check ✅ SUCCESS
StaticCheck_tag_closed_check ✅ SUCCESS
StaticCheck_markdownlint ✅ SUCCESS
Compile_Ascend_ARM_ubuntu24 ✅ SUCCESS >>>>>
Compile_Ascend_single_ubuntu24 ✅ SUCCESS >>>>>
Compile_Ascend_experimental_ubuntu24 ✅ SUCCESS >>>>>
Compile_Ascend_ARM_950_ubuntu24 ✅ SUCCESS >>>>>
UT_Test_ophost ✅ SUCCESS
UT_Test_opapi ✅ SUCCESS
UT_Test_kernel ✅ SUCCESS
UT_Test_opgraph ✅ SUCCESS
Smoke_A900_npupool ✅ SUCCESS >>>>>

[2026-03-16 14:57:58]    CI执行结束

likedislike
CANN-robotCANN-robot成员
3月16日 添加了label:api-check-pass
CANN-robotCANN-robot成员
3月16日 删除了label:ci-pipeline-running
CANN-robotCANN-robot成员
3月16日 添加了label:ci-pipeline-passed
陈展熹成员
3月17日 修改了pull request 的描述
范其瑞
范其瑞成员
3月17日 评论:

/lgtm

likedislike
chenqi317成员
3月18日 评论:

/lgtm
/approve

likedislike
CANN-robotCANN-robot成员
3月18日 添加了label:lgtmapproved
CANN-robot
CANN-robot成员
3月18日 评论:

Review Guide

This pull-request passes review.
Committers who wrote a comment of /approve are: chenqi317.
Reviewers who wrote a comment of /lgtm are: chenqi317, fanqirui.

likedislike
CANN-robotCANN-robot成员
3月18日 合入了pull request