0
代码
代码
Issues
Pull Requests
讨论
Wiki
分析
0
ascend-robotascend-robot[Feature] NPUGraph capture support compiled model
1c292736创建于 2025年11月21日历史提交
文件最后提交记录最后更新时间
aclnn
!10338 support to conv.allow_hf32 to aclnn.allow_hf32 Merge pull request !10338 from 关龙锋/cherry-pick-1710759350 2 年前
amp
update _growth_tracker with torch Co-authored-by: huangyunlong2022<huangyunlong4@h-partners.com> # message auto-generated for no-merge-commit merge: merge 2.9sg into master update _growth_tracker with torch Created-by: huangyunlong2022 Commit-by: huangyunlong2022 Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **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**: **Which issue(s) this PR fixes**: <!-- *Automatically closes linked issue when PR is merged. Usage: Fixes #<issue number>, or Fixes (paste link of issue). --> Fixes # **Special notes for your reviewers**: See merge request: Ascend/pytorch!251988 个月前
__init__.py
add interface to set op exec timeout with ms Co-authored-by: 周锐淇<zhouruiqi5@huawei.com> # message auto-generated for no-merge-commit merge: !26177 merge master into master add interface to set op exec timeout with ms Created-by: rich9527 Commit-by: 周锐淇 Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **What type of PR is this?** > /kind feature **What does this PR do / why do we need it**: This PR adds the ability to set the NPU operation timeout from Python. It allows users to configure a custom timeout for NPU operations to prevent long-running or stuck operations during execution. **Which issue(s) this PR fixes**: <!-- *Automatically closes linked issue when PR is merged. Usage: Fixes #<issue number>, or Fixes (paste link of issue). --> Fixes # **Special notes for your reviewers**: See merge request: Ascend/pytorch!261776 个月前
_backends.py
!16579 clean sdp and soc_version api Merge pull request !16579 from 关龙锋/cherry-pick-1733194542 1 年前
_fft_plan_cache.py
!17747 支持配置FFT PLAN Merge pull request !17747 from zhanglei/master 1 年前
_format.py
sync acl_format_2/4/8 Co-authored-by: 张泉新<zhangquanxin7@h-partners.com> # message auto-generated for no-merge-commit merge: !26696 merge merge_public_master into master sync acl_format_2/4/8 Created-by: zhangquanxin Commit-by: 张泉新 Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **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**: **Which issue(s) this PR fixes**: <!-- *Automatically closes linked issue when PR is merged. Usage: Fixes #<issue number>, or Fixes (paste link of issue). --> Fixes # **Special notes for your reviewers**: See merge request: Ascend/pytorch!266966 个月前
_graph_tree.py
!23238 add npugraph_tree patch, support reduce-overhead mode Merge pull request !23238 from 闫鹏全/master 10 个月前
_kernel_check.py
!16651 Rectify the common interface about sanitizer Merge pull request !16651 from hanqing/master 1 年前
_memory_viz.py
!24710 Remove unused import of subprocess Merge pull request !24710 from kuhn/master_910 8 个月前
_recovery.py
!21130 support suspect mem error and hccs link error Merge pull request !21130 from 王超/v2.7.0_linkerror 1 年前
_sanitizer.py
【bugfix】修改sanitizer的apply应用至开启流间竞争检测时 Co-authored-by: 龚昊宇<gonghaoyu1@h-partners.com> # message auto-generated for no-merge-commit merge: merge 922fix_master into master 【bugfix】修改sanitizer的apply应用至开启流间竞争检测时 Created-by: gonghaoyu Commit-by: 龚昊宇 Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **What type of PR is this?** /kind bug > 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**: 修改sanitizer的apply应用范围至开启流间竞争检测时，以防止第三方工具使用is_cuda时误判npu和cuda的tensor。 **Which issue(s) this PR fixes**: <!-- *Automatically closes linked issue when PR is merged. Usage: Fixes #<issue number>, or Fixes (paste link of issue). --> Fixes # **Special notes for your reviewers**: See merge request: Ascend/pytorch!250938 个月前
_stream_check.py
[bugfix] Add proper handling for view and factory function for csan, and supplement the corresponding test cases. Co-authored-by: chenkun<chenkun82@huawei.com> # message auto-generated for no-merge-commit merge: !26406 merge cherry-pick-mr-26401-1762829831101-auto into master [bugfix] Add proper handling for view and factory function for csan, and supplement the corresponding test cases. Created-by: kuhn7 Commit-by: chenkun Merged-by: ascend-robot Description: **What type of PR is this?** bug **What does this PR do / why do we need it**: 增加 ArgumentHander.parse_input() 缺失的 is_factory 参数， 并补充了相应的看护用例。 **Which issue(s) this PR fixes**: <!-- *Automatically closes linked issue when PR is merged. Usage: Fixes #<issue number>, or Fixes (paste link of issue). --> Fixes # **Special notes for your reviewers**: 开源社区对应改动如下： https://github.com/pytorch/pytorch/commit/c4ed03cea17d2ec9b09427786627bba479f4cb6b See merge request: Ascend/pytorch!264066 个月前
autocast_utils.py
!7315 [Feature] Autocast suppports bf16. Merge pull request !7315 from will-devil/mautocast 2 年前
backends.py
!17747 支持配置FFT PLAN Merge pull request !17747 from zhanglei/master 1 年前
deterministic.py
!13006 Fixed for the public APIs. Merge pull request !13006 from yuhaiyan/master-dev2 1 年前
graphs.py
[Feature] NPUGraph capture support compiled model Co-authored-by: XDaoHong<xudaohong@huawei.com> # message auto-generated for no-merge-commit merge: !26816 merge master into master [Feature] NPUGraph capture support compiled model Created-by: XDaoHong Commit-by: XDaoHong Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **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 > feature **What does this PR do / why do we need it**: 问题背景： torch.npu.graph(auto_dispatch_capture=True)场景，如果captruetorch.compile后的model，compile会被跳过失效。 问题原因: auto_dispatch_capture=True中会使用class _GraphDispatchMode(torch.utils._python_dispatch.TorchDispatchMode)做update算子的dispatch拦截；_GraphDispatchMode为用户自定义TorchDispatchMode，类方法def is_infra_mode(cls):默认返回False；torch.\_dynamo流程中会校验是否是用了non-infra的dispatch mode，使用的话会跳过dynamo逻辑，详见[pytorch源码](https://github.com/pytorch/pytorch/blob/v2.7.0/torch/_dynamo/convert_frame.py#L1390)。 测试用例： ```python import logging import torch import torch_npu torch._logging.set_logs(dynamo=logging.DEBUG, aot=logging.DEBUG, output_code=True, graph_code=True) class Model(torch.nn.Module): def __init__(self): super(Model, self).__init__() def forward(self, x): y = x + 1 return y model = Model().npu() model = torch.compile(model, fullgraph=True, backend="aot_eager", dynamic=False) static_input = torch.zeros((5, 5), device="npu") g = torch_npu.npu.NPUGraph() with torch.npu.graph(g, auto_dispatch_capture=True): res = model(static_input) x = torch.zeros((5, 5), device="npu") + 1 static_input.copy_(x) g.replay() print(res) ``` 关键日志： ```shell V1120 15:27:06.337000 1308894 site-packages/torch/_dynamo/convert_frame.py:1395] skipping: _wrapped_call_impl (reason: in skipfiles, file: lib/python3.11/site-packages/torch/nn/modules/module.py) V1120 15:27:06.338000 1308894 site-packages/torch/_dynamo/convert_frame.py:1395] skipping: _call_impl (reason: in skipfiles, file: lib/python3.11/site-packages/torch/nn/modules/module.py) V1120 15:27:06.338000 1308894 site-packages/torch/_dynamo/convert_frame.py:1395] skipping: forward (reason: non-infra torch dispatch mode present, this is not supported today in torch.compile, file:eager.py) ``` **Which issue(s) this PR fixes**: <!-- *Automatically closes linked issue when PR is merged. Usage: Fixes #<issue number>, or Fixes (paste link of issue). --> Fixes # **Special notes for your reviewers**: See merge request: Ascend/pytorch!268166 个月前
memory.py
check oom with error code EL0004, add AclrtGetMemUsageInfo Co-authored-by: zhaoyu<nanzhaogang@qq.com> # message auto-generated for no-merge-commit merge: !26067 merge snapshot-master into master check oom with error code EL0004, add AclrtGetMemUsageInfo Created-by: zhaoyu65 Commit-by: zhaoyu Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **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 feature **What does this PR do / why do we need it**: 1、更新对OOM场景的校验逻辑，使用EL0004错误码进行校验。 2、添加AclrtGetMemUsageInfo获取CANN内存快照 **Which issue(s) this PR fixes**: <!-- *Automatically closes linked issue when PR is merged. Usage: Fixes #<issue number>, or Fixes (paste link of issue). --> Fixes # **Special notes for your reviewers**: See merge request: Ascend/pytorch!260676 个月前
mstx.py
!20833 [PROF]mstx add domain Merge pull request !20833 from 梅飞要/domain 1 年前
npu_config.py
fix space error Co-authored-by: guoguanghao<guoguanghao@huawei.com> # message auto-generated for no-merge-commit merge: merge master into master fix space error Created-by: guoguanghao Commit-by: guoguanghao Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **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**: **Which issue(s) this PR fixes**: <!-- *Automatically closes linked issue when PR is merged. Usage: Fixes #<issue number>, or Fixes (paste link of issue). --> Fixes # **Special notes for your reviewers**: See merge request: Ascend/pytorch!249178 个月前
random.py
!6346 Resolve codecheck issue (torch_npu/csrc/libs, torch_npu/csrc/npu, torch_npu/npu) Merge pull request !6346 from wuhy/codecheck-torch_npu_npu-master 2 年前
streams.py
!20261 [feat] aclGraph task group Merge pull request !20261 from xudaohong/master 1 年前
utils.py
add interface to set op exec timeout with ms Co-authored-by: 周锐淇<zhouruiqi5@huawei.com> # message auto-generated for no-merge-commit merge: !26177 merge master into master add interface to set op exec timeout with ms Created-by: rich9527 Commit-by: 周锐淇 Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **What type of PR is this?** > /kind feature **What does this PR do / why do we need it**: This PR adds the ability to set the NPU operation timeout from Python. It allows users to configure a custom timeout for NPU operations to prevent long-running or stuck operations during execution. **Which issue(s) this PR fixes**: <!-- *Automatically closes linked issue when PR is merged. Usage: Fixes #<issue number>, or Fixes (paste link of issue). --> Fixes # **Special notes for your reviewers**: See merge request: Ascend/pytorch!261776 个月前