已合并
add torch_npu._C._npu_getCurrentRawStreamNoWait() #29737
liujunzhu创建于 1月19日
add torch_npu._C._npu_getCurrentRawStreamNoWait() #29737
已合并
liujunzhu创建于 1月19日
liujunzhu
liujunzhu成员
1月19日

What type of PR is this?

Uncomment only one /kind <> line, hit enter to put that in a new line, and remove leading whitespaces from that line:

/kind bug
/kind task

/kind feature

What does this PR do / why do we need it:
现有的torch_npu._C._npu_getCurrentRawStream()由于清空taskqueue导致性能比较差,所以添加不清空taskqueue的torch_npu._C._npu_getCurrentRawStreamNoWait()接口。
由于新增的torch_npu._C._npu_getCurrentRawStreamNoWait(device)接口不清空taskqueue,如果既使用返回的aclrtStream下发任务又使用torch_npu的taskqueue下发任务,则会引起保序问题,需由用户确保只使用其中的一种下发方式(如果混用需确保无数据依赖且性能不敏感)。

Which issue(s) this PR fixes:

Fixes #

Special notes for your reviewers:

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 liujunzhu 的贡献)
liujunzhuliujunzhu成员
1月19日 创建了 pull request,commit 913bbea4
ascend-robot
ascend-robot成员
1月19日 评论:

Thank your for your pull-request.

The full list of commands accepted by me can be found at here.

You can get sig-info at here

likedislike
ascend-robot
ascend-robot成员
1月19日 评论:

以下是根据您提交的修改文件推荐的Reviewer和Committer序列,需各模块评审通过后方可合入

Module List Reviewers Committers
repo-Ascend/pytorch wangqiang160, liwei386, guoguanghao, yuhaiyan, duchengkun wangqiang160, medivh-x, zqwenn, anyrenwei, weizhan4
likedislike
ascend-robotascend-robot成员
1月19日 添加了label:ascend-cla/yes
ascend-robot
ascend-robot成员
1月19日 评论:

CLA Signature Pass

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

likedislike
liujunzhu
liujunzhu成员
1月19日 评论:

add_component

likedislike
ascend-robotascend-robot成员
1月19日 添加了label:MindSpore-Component
liujunzhu
liujunzhu成员
1月19日 评论:

compile

likedislike
ascend-robotascend-robot成员
1月19日 添加了label:ci-pipeline-running
AtlasAccount
AtlasAccount成员
1月19日 评论:

/torch_npu/csrc/npu/Module.cpp

likedislike
ascend-robotascend-robot成员
1月19日 删除了label:ci-pipeline-running
ascend-robotascend-robot成员
1月19日 添加了label:ci-pipeline-passed
ascend-robot
ascend-robot成员
1月19日 评论:
流水线 PR-pipeline_pytorch#3745 已完成
阶段 任务名 状态 详情
编译构建 Build_X86 >>>
Build_LibTorch >>>
Build_ARM >>>
Build_ARM_inductor 🛑 >>>
Build_X86_py311 🛑 >>>
Build_ARM_py311 🛑 >>>
dist_test_or_not >>>
恶意代码检查 Antipoison >>>
编码安全与规范检查 CodeCheck >>>
check_error >>>
开源片段检查 SCA >>>
开发者测试 UT_DIST_X86 🛑 >>>
UT_X86_Part_01 >>>
UT_X86_Part_02 >>>
UT_ARM_A2_Part_01 🛑 >>>
UT_ARM_A2_Part_02 🛑 >>>
UT_inductor_arm 🛑 >>>
流水线 PR-pipeline_pytorch >>>
likedislike
王超成员
1月28日 评论:

/approve

likedislike
ascend-robotascend-robot成员
1月28日 添加了label:lgtm
li_jing_hw成员
1月28日 评论:

/approve

likedislike
ascend-robotascend-robot成员
1月28日 添加了label:approved
ascend-robot
ascend-robot成员
1月28日 评论:

Review Guide

This Pull-Request Passes Review.
Committers who wrote a comment of /approve are: wangchao430, li_jing_hw.
Reviewers who wrote a comment of /lgtm are: wangchao430, li_jing_hw.

likedislike
ascend-robotascend-robot成员
1月28日 合入了pull request