| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 11 个月前 | ||
add control of python GC before capture npugraph Co-authored-by: 周锐淇<zhouruiqi5@huawei.com> # message auto-generated for no-merge-commit merge: !27714 merge v2.8.0 into v2.8.0 add control of python GC before capture npugraph Created-by: rich9527 Commit-by: 周锐淇 Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **What type of PR is this?** > /kind task **What does this PR do / why do we need it**: add control of python GC before capture npugraph **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!27714 | 8 个月前 | |
update: 更新文件 README.md Co-authored-by: cuiduo<cuiduo1@huawei.com> # message auto-generated for no-merge-commit merge: !31104 merge v2.8.0 into v2.8.0 update: 更新文件 README.md Created-by: cuiduo Commit-by: cuiduo Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">**如有社区issue，请关联issue链接**</font>\ > <font color="red">**请勿携带内部流程信息（需求链接、问题单、内部issue等）**</font> - [ ] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [x] 资料更新 # 【修改方案】 > 新增mlir的使用方法描述资料，位于torch_npu/_inductor/ascend_npu_ir/README.md中 # 【资料变更】 > 在torch_npu/_inductor/ascend_npu_ir/README.md中简要阐述了mlir的基本信息与使用方法 # 【接口变更】 >不涉及 # 【功能验证】 > 已本地验证通过 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!31104 | 6 个月前 | |
[fa_v3]skip cpu check, register_sharding for npu_fusion_attention_v3 Co-authored-by: wangchao430<wangchao430@huawei.com> # message auto-generated for no-merge-commit merge: !30859 merge v2.8.0_fav3 into v2.8.0 [fa_v3]skip cpu check, register_sharding for npu_fusion_attention_v3 Created-by: wangchao430 Commit-by: wangchao430 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">**如有社区issue，请关联issue链接**</font>\ > <font color="red">**请勿携带内部流程信息（需求链接、问题单、内部issue等）**</font> - [x] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 1. torch_npu.npu_fusion_attention_v3适配支持DTensor策略注册。 2. graph_tree中支持根据torch_npu.\_inductor.config.npugraph_trees.disable_cpu_input_check配置开启时跳过cpu Tensor的相关检测，从而支持torch_npu.npu_fusion_attention_v3入graph_tree # 【资料变更】 不涉及 # 【接口变更】 不涉及 # 【功能验证】 新增DTensor测试用例 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!30859 | 6 个月前 | |
[SilentCheck] fix with DTensor of grad Co-authored-by: wangchao430<wangchao430@huawei.com> # message auto-generated for no-merge-commit merge: merge v2.8.0_checkfix1 into v2.8.0 [SilentCheck] fix with DTensor of grad Created-by: wangchao430 Commit-by: wangchao430 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!25007 | 11 个月前 | |
fix bug in transfer_to_npu Co-authored-by: kkjocker<hexuanyu1@huawei.com> # message auto-generated for no-merge-commit merge: !30009 merge v2.8.0 into v2.8.0 fix bug in transfer_to_npu Created-by: kkjocker Commit-by: kkjocker 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**: 原本有些代码和torch.jit.script不适配，使用try except方式使能torch.jit.script后会引入别的bug，本pr通过use_jit_script的方式来使能torch.jit.script > import torch_npu > from torch_npu.utils import _dynamo > _dynamo.use_jit_script = True > from torch_npu.contrib import transfer_to_npu **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!30009 | 6 个月前 | |
Add PREMUL_SUM support in ProcessGroupHCCL for ReduceOp Co-authored-by: dairenjie<dairenjie@huawei.com> # message auto-generated for no-merge-commit merge: !31240 merge v2.8.0_0303 into v2.8.0 Add PREMUL_SUM support in ProcessGroupHCCL for ReduceOp Created-by: dairenjie Commit-by: dairenjie Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 - [x] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 1. 修改ProcessGroupHCCL内reduce类接口的实现，在pre函数添加input.mul_(factor)的调用（9处） 2. 修改getHcclReduceOp内处理premul_sum场景为sum # 【资料变更】 不涉及 # 【接口变更】 增加torch.distributed._make_hccl_premul_sum 和PT原生接口_make_nccl_premul_sum保持对等，已评审。 # 【功能验证】 修改/新增reduce算子premul_sum场景测试用例。 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!31240 | 5 个月前 | |
Add PREMUL_SUM support in ProcessGroupHCCL for ReduceOp Co-authored-by: dairenjie<dairenjie@huawei.com> # message auto-generated for no-merge-commit merge: !31240 merge v2.8.0_0303 into v2.8.0 Add PREMUL_SUM support in ProcessGroupHCCL for ReduceOp Created-by: dairenjie Commit-by: dairenjie Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 - [x] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 1. 修改ProcessGroupHCCL内reduce类接口的实现，在pre函数添加input.mul_(factor)的调用（9处） 2. 修改getHcclReduceOp内处理premul_sum场景为sum # 【资料变更】 不涉及 # 【接口变更】 增加torch.distributed._make_hccl_premul_sum 和PT原生接口_make_nccl_premul_sum保持对等，已评审。 # 【功能验证】 修改/新增reduce算子premul_sum场景测试用例。 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!31240 | 5 个月前 | |
convolution-meta | 5 个月前 | |
| 2 年前 | ||
Event supports cross-process and cross-device (IPC event) Co-authored-by: liujunzhu<liujunzhu@huawei.com> # message auto-generated for no-merge-commit merge: !28521 merge v2.8.0 into v2.8.0 Event supports cross-process and cross-device (IPC event) Created-by: liujunzhu Commit-by: liujunzhu 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**: event能力对齐cuda，支持跨进程与跨设备使用。 CUDA跨进程共享内存和跨设备拷贝内存时使用Event进行同步，而torch npu使用SynchronizeStream进行同步，希望torch_npu支持在跨设备和跨内存的场景也使用Event进行同步，提升整体性能。另外，需支持通过Python接口在进程间传递Event对象或Event handle并使用该Event进行进程间的同步。 相关场景如下： 1、跨进程使用Event：进程间传递interprocess=True的Event对象、Event的IPC Handle或torch.multiprocessing.reductions.reduce_event(event)的结果。 2、跨进程共享NPU内存：包括进程间通过参数或队列传递Tensor、进程间通过torch.multiprocessing.reductions.reduce_tensor()传递Tensor、进程间通过_share_npu_传递Tensor。 3、跨设备拷贝NPU内存：调用Tensor.to()或Tensor.copy_()接口。 不支持跨设备或跨进程使用interprocess=False的Event。 **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!28521 | 8 个月前 | |
[fa_v3]skip cpu check, register_sharding for npu_fusion_attention_v3 Co-authored-by: wangchao430<wangchao430@huawei.com> # message auto-generated for no-merge-commit merge: !30859 merge v2.8.0_fav3 into v2.8.0 [fa_v3]skip cpu check, register_sharding for npu_fusion_attention_v3 Created-by: wangchao430 Commit-by: wangchao430 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">**如有社区issue，请关联issue链接**</font>\ > <font color="red">**请勿携带内部流程信息（需求链接、问题单、内部issue等）**</font> - [x] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 1. torch_npu.npu_fusion_attention_v3适配支持DTensor策略注册。 2. graph_tree中支持根据torch_npu.\_inductor.config.npugraph_trees.disable_cpu_input_check配置开启时跳过cpu Tensor的相关检测，从而支持torch_npu.npu_fusion_attention_v3入graph_tree # 【资料变更】 不涉及 # 【接口变更】 不涉及 # 【功能验证】 新增DTensor测试用例 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!30859 | 6 个月前 | |
rename 910_95->950 Co-authored-by: MrMC-<shiqunze@h-partners.com> # message auto-generated for no-merge-commit merge: !29766 merge 2.8.0-ad-950 into v2.8.0 rename 910_95->950 Created-by: MrMC- Commit-by: MrMC- 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!29766 | 7 个月前 | |
| 2 年前 | ||
[fix]profiler db analyse optimize Co-authored-by: hehongzhe<935062458@qq.com> # message auto-generated for no-merge-commit merge: !30647 merge jx8 into v2.8.0 [fix]profiler db analyse optimize Created-by: hhz0 Commit-by: hehongzhe Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > (如有)请关联需求文档/issue链接 - [ ] 需求 - [x] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 > 请描述修改内容的具体实现，涉及哪些组件之间进行交互，可以用1、2、3、...进行罗列\ > 如果是需求或者重构类的PR，需要补充详细设计文档（说明上下游组件关系、时序图、类图、DFX能力等内容） 在db解析的情况下node_launch与enqueue匹配或是torchop与enqueue匹配的时候，进行双指针匹配，但是匹配过程中，没有对未匹配上的场景进行判断，导致双指针失效。 本pr新增了对未匹配场景的指针更新与break逻辑。 # 【资料变更】 > 请确认是否涉及资料变更。如涉及，需要在PR中体现，并简要说明修改内容。如不涉及，需填写“不涉及” 不涉及 # 【接口变更】 > 请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及，需要详细说明接口以及对应的变更内容，同时需要在资料中体现。如不涉及，需填写“不涉及” 不涉及 # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图 对于同一份数据，修改前后，db解析的时长有明显变短   # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!30647 | 6 个月前 | |
Fix __rmod__ testcase. Co-authored-by: liangchenghui<liangchenghui@huawei.com> # message auto-generated for no-merge-commit merge: !30902 merge fix___rmod___testcase_2.8.0 into v2.8.0 Fix __rmod__ testcase. Created-by: liangchenghui Commit-by: liangchenghui Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > (如有)请关联需求文档/issue链接 - [ ] 需求 - [ ] 问题单 - [x] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 ## 现象 - 用例 test_ops.py::test_dtypes___rmod___npu 失败。 - 报错指向：前向实际可用的 dtype（torch.int32, torch.int64）未在 OpInfo 中登记。 ## 责任模块 - 主要：torch_npu.testing 的 OpInfo 适配/注册逻辑（NPU 测试基础设施） - 关联：上游 PyTorch 的 OpInfo dtype 定义（__rmod__ 的默认 dtype 列表） ## 根因分析 - NPU 侧 __rmod__ 前向对 int32/int64 实际可用，但测试侧 OpInfo 仍沿用上游默认 dtype 列表，未包含这两种整型。 - NPU 测试适配层对 OpInfo 的 supported_dtypes() 直接透传 self.dtypes，导致前向实际能力与登记信息不一致。 ## 解决方案 ### 已实施（配置驱动） - 在 torch_npu/testing/__init__.py 中引入配置驱动的 dtype 扩展： - 新增 torch_npu/testing/npu_opinfo_dtypes.py，按 op 维护额外 dtype。 - supported_dtypes() 读取配置并合并，避免硬编码。 # 【资料变更】 不涉及 # 【接口变更】 不涉及 # 【功能验证】 修改已有测试用例。 # 【CheckList】 - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!30902 | 6 个月前 | |
[fa_v3]skip cpu check, register_sharding for npu_fusion_attention_v3 Co-authored-by: wangchao430<wangchao430@huawei.com> # message auto-generated for no-merge-commit merge: !30859 merge v2.8.0_fav3 into v2.8.0 [fa_v3]skip cpu check, register_sharding for npu_fusion_attention_v3 Created-by: wangchao430 Commit-by: wangchao430 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">**如有社区issue，请关联issue链接**</font>\ > <font color="red">**请勿携带内部流程信息（需求链接、问题单、内部issue等）**</font> - [x] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 1. torch_npu.npu_fusion_attention_v3适配支持DTensor策略注册。 2. graph_tree中支持根据torch_npu.\_inductor.config.npugraph_trees.disable_cpu_input_check配置开启时跳过cpu Tensor的相关检测，从而支持torch_npu.npu_fusion_attention_v3入graph_tree # 【资料变更】 不涉及 # 【接口变更】 不涉及 # 【功能验证】 新增DTensor测试用例 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!30859 | 6 个月前 | |
[feat] add env TORCH_TRANSFER_TO_NPU Co-authored-by: liujunzhu<liujunzhu@huawei.com> # message auto-generated for no-merge-commit merge: !30827 merge v2.8.0 into v2.8.0 [feat] add env TORCH_TRANSFER_TO_NPU Created-by: liujunzhu Commit-by: liujunzhu Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > (如有)请关联需求文档/issue链接 - [x] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 > 请描述修改内容的具体实现，涉及哪些组件之间进行交互，可以用1、2、3、...进行罗列\ > 如果是需求或者重构类的PR，需要补充详细设计文档（说明上下游组件关系、时序图、类图、DFX能力等内容） 支持通过环境变量 TORCH_TRANSFER_TO_NPU使能transfer_to_npu。 设置环境变量TORCH_TRANSFER_TO_NPU为1时（取值为0或1），在import torch_npu时自动执行from torch_npu.contrib import transfer_to_npu，方便Torch CUDA脚本无缝迁移到NPU。 # 【资料变更】 > 请确认是否涉及资料变更。如涉及，需要在PR中体现，并简要说明修改内容。如不涉及，需填写“不涉及” 涉及资料变更，资料中需描述TORCH_TRANSFER_TO_NPU的功能和用法，本pr不涉及资料修改，后续单独提交资料修改的pr。 # 【接口变更】 > 请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及，需要详细说明接口以及对应的变更内容，同时需要在资料中体现。如不涉及，需填写“不涉及” 新增环境变量TORCH_TRANSFER_TO_NPU，取值为0或1，默认为0，设置为1时支持cuda接口到npu的转换。 # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图 PR已附带大量验证用例。 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!30827 | 6 个月前 | |
| 1 年前 | ||
| 11 个月前 |