已合并
【feat】mstx support push and pop #36207
mei-feiyao创建于 5月20日
【feat】mstx support push and pop #36207
已合并
Pull Request已成功合入, 合并人@ascend-robot
（感谢 mei-feiyao 的贡献）5月20日 创建了 pull request，commit adbd362a
ascend-robot
5月20日 评论：
5月20日 评论：
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 |
|---|---|---|
| test | ✅ 陈豪, 李伟, liujunzhu (3/2) | ✅ 陈豪, 李伟 (2/1) |
| torch_npu/csrc/profiler | ✅ 陈豪, 李伟, liujunzhu (3/2) | ✅ 陈豪, 李伟 (2/1) |
| torch_npu/npu | ✅ 李伟, liujunzhu (2/2) | ✅ 李伟 (1/1) |
| torch_npu/profiler | ✅ 陈豪, 李伟, liujunzhu (3/2) | ✅ 李伟, 陈豪 (2/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
mei-feiyao, thanks for your pull request. All authors of the commits have signed the CLA. 👍
5月20日 添加了label：ascend-cla/yes
mei-feiyao
5月20日 评论：
5月20日 评论：
compile
ascend-robot
5月20日 评论：
5月20日 评论：
当前仓库存在以下 保护分支 ：
| Protected Branch | Version | Release |
|---|---|---|
| master | ||
| v2.11.0 | ||
| v2.10.0 | ||
| v2.7.1 | ||
| v2.9.0 | ||
| v2.12.0 |
评论 /sync <branch1> <branch2> ... 可将当前 PR 修改同步到其它分支（创建同步 PR）：
a) 如果当前 PR 是 Open 状态，同步操作将延迟到 PR 被合并时执行
b) 如果当前 PR 已经 Merged，将立即执行同步操作
注意：
- /sync 命令可以指定同步到多个分支，仅最后一个 /sync 命令生效
- 如果创建的同步 PR 不正确，可通过向同步 PR 的源分支提交轻量级 PR 完善，或使用 /close 命令关闭
此处折叠了176条消息 查看更多
5月21日 添加了label：approved
liujunzhu
5月21日 评论：
5月21日 评论：
/lgtm
5月21日 添加了label：lgtm
5月21日 合入了pull request
【合入来源】
【修改方案】
对齐社区nvtx模块，补全接口，包括如下：
1、range_push、range_pop：线程粒度的打range点接口，内部走mstx库实现打点，用户可通过torch_npu.profiler接口配置mstx=True或者msproftx=True(待废弃)的方式采集tx打点数据；
2、annotate类：用户可通过with torch_npu.npu.mstx.annotate(message="xxxx", stream=xxxx, domain="tttt")的方式或者@torch_npu.npu.mstx.annotate(message="xxxx", stream=xxxx, domain="tttt")装饰器的方式对目标代码段或者目标函数进行打点，内部会在开始和结束的阶段分别调用range_push和range_pop
【资料变更】
是。涉及新增接口，已联系资料同学补充资料
【接口变更】
本需求只涉及新增接口，不涉及已有接口变更
【功能验证】
1、UT测试验证新增接口；
2、模型代码增加调用新增的打点接口，并通过profiler采集打点数据，观察打点数据正常
【CheckList】