pytorch_6658/test · cuiduo/pytorch_6658

文件	最后提交记录	最后更新时间
_afd	skip ci failed testcase Co-authored-by: XianglongZeng<zengxianglong1@huawei.com> # message auto-generated for no-merge-commit merge: !36277 merge v2.12.0 into v2.12.0 skip ci failed testcase Created-by: probiotics_53 Commit-by: XianglongZeng Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 https://gitcode.com/Ascend/op-plugin/pull/4958合入受阻，下掉一批与此次修改无关的门禁失败用例： ## 1. `_afd/test_attention_worker_schedule` `File "/workspace/CODE/pytorch_ut/test/_afd/test_attention_worker_schedule.py", line 4, in <module> import torchair ModuleNotFoundError: No module named 'torchair' [ERROR] ERR99999 UNKNOWN applicaiton exception` ## 2. `_afd/test_ffn_worker_schedule` 同上：`ModuleNotFoundError: No module named 'torchair'` + `ERR99999 UNKNOWN` ## 3. `custom_ops/test_npu_multi_head_attention` `File "/workspace/CODE/pytorch_ut/test/custom_ops/test_npu_multi_head_attention.py", line 98 self.assertRtolEqual(cpu_result, npu_result) AssertionError: result error` 精度比较失败（NPU 和 CPU 计算结果差异超阈值）。 ## 4. `distributed/pipelining/test_schedule`（2 个 error） `ERROR [0.288s]: test_grad_with_split_b_w schedule._load_actions(...) AttributeError: '_PipelineScheduleRuntime' object has no attribute '_load_actions' ERROR [0.058s]: test_grad_with_v_schedule torch.distributed.init_process_group(...)` `_PipelineScheduleRuntime` API 缺失 `_load_actions` 方法，distributed 接口不兼容。 ## 5. `nn/test_module_hooks` `File "/workspace/CODE/pytorch_ut/test/nn/test_module_hooks.py", line 1487 with self.assertRaisesRegex(RuntimeError, "Output 0 of BackwardHookFunctionBackward is " AssertionError: expected: "Output 0 of BackwardHookFunctionBackward is a view..." got: "Output 0 of BackwardHookFunction is a view..."` PyTorch 报错信息中类名变更（`BackwardHookFunctionBackward` → `BackwardHookFunction`），正则匹配失败。 ## 6. `test_multiprocessing_api` `File "/workspace/CODE/pytorch_ut/test/test_multiprocessing_api.py", line 215 self.assertTrue(torch.equal(npu_tensor.cpu(), reconstructed_npu.cpu())) AssertionError: False is not true` 多进程 tensor 序列化/反序列化后数据不一致。 ## 7. `test_ops _complex`（3 个 error，同一根因） `RuntimeError: _index_put_impl_:IndexPutKernelNpuOpApi.cpp:82 call aclnnIndexPutImpl failed, error code is 161002 AclNN_Parameter_Error(EZ1001): Tensor self not implemented for DT_COMPLEX64, should be in dtype support list [DT_FLOAT, DT_INT32, DT_INT64, ...]` ACL 底层 `aclnnIndexPutImpl` 不支持 complex64/complex128 数据类型。 ## 总结 \| # \| 测试 \| 根因 \| 与 0d CPU tensor 修改关系 \| \|---\|------\|------\|:--:\| \| 1 \| attention_worker_schedule \| 环境缺 `torchair` 包 \| 无关 \| \| 2 \| ffn_worker_schedule \| 环境缺 `torchair` 包 \| 无关 \| \| 3 \| multi_head_attention \| 精度差异 \| 无关 \| \| 4 \| test_schedule \| distributed API `_load_actions` 缺失 \| 无关 \| \| 5 \| module_hooks \| PyTorch 错误消息类名变更 \| 无关 \| \| 6 \| multiprocessing_api \| 多进程 tensor 序列化不一致 \| 无关 \| \| 7 \| test_ops _complex \| ACL 不支持 complex dtype \| 无关 \| https://gitcode.com/Ascend/op-plugin/pull/4958 涉及的 15 个 `test_base_ops/` 测试全部 success。 # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】不涉及 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!36277	3 个月前
_inductor	wrap_triton	3 个月前
allocator	HOST内存统计API对齐社区 Co-authored-by: luanchaowei<luanchaowei3@huawei.com> # message auto-generated for no-merge-commit merge: !29057 merge master into master HOST内存统计API对齐社区 Created-by: luanchaowei Commit-by: luanchaowei 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: 1.torch_npu.npu. empty_pin_memory_cache()更名为torch_npu.npu.host_empty_cache() 2.torch_npu.npu.pin_memory_stats()更名为torch_npu.npu.host_memory_stats() 3.删除torch_npu.npu.pin_memory_allocated()和torch_npu.npu.pin_memory_reserved() 4.补齐torch_npu.npu.host_memory_stats()其他指标 5.新增 torch_npu.npu.host_memory_stats_as_nested_dict() torch_npu.npu.reset_accumulated_host_memory_stats() torch_npu.npu. reset_peak_host_memory_stats() 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!29057	7 个月前
autograd	fix: lintrunner --all-files --take NEWLINE -a Co-authored-by: huangjingwei<huangjingwei4@huawei.com> # message auto-generated for no-merge-commit merge: !35875 merge v2.12.0_lintrunner into v2.12.0 fix: lintrunner --all-files --take NEWLINE -a Created-by: huangjingwei Commit-by: huangjingwei Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 1. 执行：`lintrunner --all-files --take NEWLINE -a` 修复换行符，不影响代码逻辑。 2. 屏蔽两个分布式用例，已提issue，给对应接口人跟踪。 # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】不涉及 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!35875	3 个月前
bottleneck_test	!9500 The test cases support NPU. Merge pull request !9500 from yuhaiyan/master-dev4	2 年前
contrib	[Feature] transfer_to_npu adapt cuda_default_generators Co-authored-by: l30044004<louyujing@huawei.com> # message auto-generated for no-merge-commit merge: !34215 merge master into master [Feature] transfer_to_npu adapt cuda_default_generators Created-by: louyujing Commit-by: l30044004 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> https://gitcode.com/Ascend/pytorch/issues/1789 - [x] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 > 请描述修改内容的具体实现，涉及哪些组件之间进行交互，可以用1、2、3、...进行罗列\ > 如果是需求或者重构类的PR，需要补充详细设计文档（说明上下游组件关系、时序图、类图、DFX能力等内容） torch.cuda.default_generators在NPU上需要迁移为torch_npu.npu.default_generators，且必须是设备init完成之后使用torch_npu.npu._lazy_call的方式，在设备init完成之后，才使patch生效 # 【资料变更】 > 请确认是否涉及资料变更。如涉及，需要在PR中体现，并简要说明修改内容。如不涉及，需填写“不涉及” 不涉及 # 【接口变更】 > 请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及，需要详细说明接口以及对应的变更内容，同时需要在资料中体现。如不涉及，需填写“不涉及” 不涉及 # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!34215	4 个月前
cpp	[Feature]rename head file and add cmake for libtorch_npu	2 年前
cpp_extensions	feat support device parameters for aoti test Co-authored-by: huangyunlong2022<huangyunlong4@h-partners.com> # message auto-generated for no-merge-commit merge: !36117 merge 2.12shim_td into v2.12.0 feat support device parameters for aoti test Created-by: huangyunlong2022 Commit-by: huangyunlong2022 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [x] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 > 请描述修改内容的具体实现，涉及哪些组件之间进行交互，可以用1、2、3、...进行罗列\ > 如果是需求或者重构类的PR，需要补充详细设计文档（说明上下游组件关系、时序图、类图、DFX能力等内容）包含device的算子用例支持传device # 【资料变更】 > 请确认是否涉及资料变更。如涉及，需要在PR中体现，并简要说明修改内容。如不涉及，需填写“不涉及” # 【接口变更】 > 请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及，需要详细说明接口以及对应的变更内容，同时需要在资料中体现。如不涉及，需填写“不涉及” # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!36117	3 个月前
custom_ops	skip ci failed testcase Co-authored-by: XianglongZeng<zengxianglong1@huawei.com> # message auto-generated for no-merge-commit merge: !36277 merge v2.12.0 into v2.12.0 skip ci failed testcase Created-by: probiotics_53 Commit-by: XianglongZeng Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 https://gitcode.com/Ascend/op-plugin/pull/4958合入受阻，下掉一批与此次修改无关的门禁失败用例： ## 1. `_afd/test_attention_worker_schedule` `File "/workspace/CODE/pytorch_ut/test/_afd/test_attention_worker_schedule.py", line 4, in <module> import torchair ModuleNotFoundError: No module named 'torchair' [ERROR] ERR99999 UNKNOWN applicaiton exception` ## 2. `_afd/test_ffn_worker_schedule` 同上：`ModuleNotFoundError: No module named 'torchair'` + `ERR99999 UNKNOWN` ## 3. `custom_ops/test_npu_multi_head_attention` `File "/workspace/CODE/pytorch_ut/test/custom_ops/test_npu_multi_head_attention.py", line 98 self.assertRtolEqual(cpu_result, npu_result) AssertionError: result error` 精度比较失败（NPU 和 CPU 计算结果差异超阈值）。 ## 4. `distributed/pipelining/test_schedule`（2 个 error） `ERROR [0.288s]: test_grad_with_split_b_w schedule._load_actions(...) AttributeError: '_PipelineScheduleRuntime' object has no attribute '_load_actions' ERROR [0.058s]: test_grad_with_v_schedule torch.distributed.init_process_group(...)` `_PipelineScheduleRuntime` API 缺失 `_load_actions` 方法，distributed 接口不兼容。 ## 5. `nn/test_module_hooks` `File "/workspace/CODE/pytorch_ut/test/nn/test_module_hooks.py", line 1487 with self.assertRaisesRegex(RuntimeError, "Output 0 of BackwardHookFunctionBackward is " AssertionError: expected: "Output 0 of BackwardHookFunctionBackward is a view..." got: "Output 0 of BackwardHookFunction is a view..."` PyTorch 报错信息中类名变更（`BackwardHookFunctionBackward` → `BackwardHookFunction`），正则匹配失败。 ## 6. `test_multiprocessing_api` `File "/workspace/CODE/pytorch_ut/test/test_multiprocessing_api.py", line 215 self.assertTrue(torch.equal(npu_tensor.cpu(), reconstructed_npu.cpu())) AssertionError: False is not true` 多进程 tensor 序列化/反序列化后数据不一致。 ## 7. `test_ops _complex`（3 个 error，同一根因） `RuntimeError: _index_put_impl_:IndexPutKernelNpuOpApi.cpp:82 call aclnnIndexPutImpl failed, error code is 161002 AclNN_Parameter_Error(EZ1001): Tensor self not implemented for DT_COMPLEX64, should be in dtype support list [DT_FLOAT, DT_INT32, DT_INT64, ...]` ACL 底层 `aclnnIndexPutImpl` 不支持 complex64/complex128 数据类型。 ## 总结 \| # \| 测试 \| 根因 \| 与 0d CPU tensor 修改关系 \| \|---\|------\|------\|:--:\| \| 1 \| attention_worker_schedule \| 环境缺 `torchair` 包 \| 无关 \| \| 2 \| ffn_worker_schedule \| 环境缺 `torchair` 包 \| 无关 \| \| 3 \| multi_head_attention \| 精度差异 \| 无关 \| \| 4 \| test_schedule \| distributed API `_load_actions` 缺失 \| 无关 \| \| 5 \| module_hooks \| PyTorch 错误消息类名变更 \| 无关 \| \| 6 \| multiprocessing_api \| 多进程 tensor 序列化不一致 \| 无关 \| \| 7 \| test_ops _complex \| ACL 不支持 complex dtype \| 无关 \| https://gitcode.com/Ascend/op-plugin/pull/4958 涉及的 15 个 `test_base_ops/` 测试全部 success。 # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】不涉及 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!36277	3 个月前
distributed	test(distributed): add test for WriteItem.tensor_storage_size Co-authored-by: qq_65095566<1922937030@qq.com> # message auto-generated for no-merge-commit merge: !35401 merge test-writeitem-tensor-storage-size-v2.12.0 into v2.12.0 test(distributed): add test for WriteItem.tensor_storage_size Created-by: qq_65095566 Commit-by: qq_65095566 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">[[Usage]: API一致性说明：torch.distributed.checkpoint.planner.WriteItem.tensor_storage_size 缺少专项验证用例 #1909](https://gitcode.com/Ascend/pytorch/issues/1909)</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [x] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 > 请描述修改内容的具体实现，涉及哪些组件之间进行交互，可以用1、2、3、...进行罗列\ > 如果是需求或者重构类的PR，需要补充详细设计文档（说明上下游组件关系、时序图、类图、DFX能力等内容）本 PR 针对 `torch.distributed.checkpoint.planner.WriteItem.tensor_storage_size` 新增专项验证用例。一、API 功能说明 `torch.distributed.checkpoint.planner.WriteItem.tensor_storage_size` 是 distributed checkpoint planner 中 `WriteItem` 提供的元数据辅助方法，用于返回当前写入项对应 tensor 的存储大小，单位为 bytes。该 API 的核心行为如下： 1. 当 `WriteItem.tensor_data` 为 `None` 时，表示该写入项不是 tensor 写入项，例如 `WriteItemType.BYTE_IO`，此时返回 `None`。 2. 当 `WriteItem.tensor_data` 不为 `None` 时，根据 `tensor_data.size` 计算元素数量，并结合 `tensor_data.properties.dtype` 计算单个元素字节数，最终返回 tensor 存储字节数。 3. 该 API 不执行实际 tensor 数值计算，不涉及 NPU kernel 或底层算子调用，主要依赖 checkpoint planner 中已有的 tensor 元数据信息。二、用例补齐说明经检索，当前测试目录中未发现直接调用 `WriteItem.tensor_storage_size()` 并断言其返回值的专项测试用例。已有 checkpoint planner 相关测试主要覆盖 `WriteItem`、`SavePlan`、`TensorWriteData` 等结构和 planner 流程，但没有单独验证 `tensor_storage_size` 的返回值行为。因此，本 PR 新增测试文件： test/distributed/checkpoint/test_planner_api.py 该测试文件覆盖以下场景： WriteItemType.TENSOR + tensor_data 构造普通 tensor 写入项，覆盖 torch.float32、torch.float16 和 torch.int8 三种 dtype，验证返回值等于： tensor.numel() * tensor.element_size() WriteItemType.SHARD + tensor_data 构造 shard 写入项，验证 tensor_storage_size 能够根据 shard 对应的 tensor metadata 返回正确的存储字节数。 WriteItemType.BYTE_IO 且 tensor_data 为 None 构造非 tensor 写入项，验证 tensor_storage_size 返回 None。三、NPU 适配说明 torch.distributed.checkpoint.planner.WriteItem.tensor_storage_size 本身是元数据辅助方法，不直接依赖 NPU 算子。但测试中需要基于 tensor 构造 TensorWriteData 和 TensorProperties，因此测试 tensor 需要运行在 NPU 设备上。本 PR 使用如下方式获取当前 accelerator： device_type = acc.type if (acc := torch.accelerator.current_accelerator()) else "cpu" 并通过如下方式将测试 tensor 迁移到当前设备： tensor = torch.empty((2, 3), dtype=dtype).to(device_type) 因此，在 NPU 环境下，测试 tensor 会运行在 NPU 设备上，满足 NPU 适配要求。四、涉及版本说明本次用例补齐分别提交到以下目标分支： v2.7.1 v2.9.0 v2.10.0 v2.11.0 v2.12.0 master 各版本均单独提交 PR，目标分支与源码分支一一对应。 # 【资料变更】 > 请确认是否涉及资料变更。如涉及，需要在PR中体现，并简要说明修改内容。如不涉及，需填写“不涉及” 涉及资料支持情况检查。本 PR 不直接修改资料文件。已检查 v2.7.1 分支 `docs/zh/native_apis`，当前文档中已有 `torch.distributed.checkpoint.planner.WriteItem`，但未检索到 `torch.distributed.checkpoint.planner.WriteItem.tensor_storage_size` 的方法级支持信息。已单独提交资料补齐 PR 到 v2.7.1 分支，补充 `docs/zh/native_apis` 中该 API 的支持情况。资料 PR 刷新以下版本目录： 1. `docs/zh/native_apis/pytorch_2-7-1/torch-distributed-checkpoint.md` 2. `docs/zh/native_apis/pytorch_2-9-0/torch-distributed-checkpoint.md` 3. `docs/zh/native_apis/pytorch_2-10-0/torch-distributed-checkpoint.md` `pytorch_2-6-0` 和 `pytorch_2-8-0` 按要求不刷新。资料补齐 PR：<https://gitcode.com/Ascend/pytorch/pull/35673> # 【接口变更】 > 请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及，需要详细说明接口以及对应的变更内容，同时需要在资料中体现。如不涉及，需填写“不涉及” 不涉及 # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图新增测试用例： test/distributed/checkpoint/test_planner_api.py 测试命令： python test/distributed/checkpoint/test_planner_api.py -v 测试场景： WriteItemType.TENSOR + tensor_data，验证返回 tensor 存储字节数。 WriteItemType.SHARD + tensor_data，验证返回 shard tensor 存储字节数。 WriteItemType.BYTE_IO 且 tensor_data 为 None，验证返回 None。已在以下版本执行该用例： v2.7.1：通过 v2.9.0：通过 v2.10.0：通过 v2.11.0：通过 v2.12.0：通过 master：通过不同版本运行用例结果示例： V2.7.1 [root@734774725e04 workspace]# python /workspace/ascend-pytorch-v2.7.1/test/distributed/checkpoint/test_planner_api.py -v test_write_item_tensor_storage_size_for_non_tensor (__main__.TestPlannerAPI.test_write_item_tensor_storage_size_for_non_tensor) ... ok test_write_item_tensor_storage_size_for_shard (__main__.TestPlannerAPI.test_write_item_tensor_storage_size_for_shard) ... ok test_write_item_tensor_storage_size_for_tensor (__main__.TestPlannerAPI.test_write_item_tensor_storage_size_for_tensor) ... ok ---------------------------------------------------------------------- Ran 3 tests in 4.995s OK [root@734774725e04 workspace]# V2.9.0 [root@734774725e04 workspace]# python /workspace/ascend-pytorch-v2.9.0/test/distributed/checkpoint/test_planner_api.py -v test_write_item_tensor_storage_size_for_non_tensor (__main__.TestPlannerAPI.test_write_item_tensor_storage_size_for_non_tensor) ... ok test_write_item_tensor_storage_size_for_shard (__main__.TestPlannerAPI.test_write_item_tensor_storage_size_for_shard) ... ok test_write_item_tensor_storage_size_for_tensor (__main__.TestPlannerAPI.test_write_item_tensor_storage_size_for_tensor) ... ok ---------------------------------------------------------------------- Ran 3 tests in 1.023s OK [root@734774725e04 workspace]# V2.10.0 [root@734774725e04 workspace]# python /workspace/ascend-pytorch-v2.10.0/test/distributed/checkpoint/test_planner_api.py -v test_write_item_tensor_storage_size_for_non_tensor (__main__.TestPlannerAPI.test_write_item_tensor_storage_size_for_non_tensor) ... ok test_write_item_tensor_storage_size_for_shard (__main__.TestPlannerAPI.test_write_item_tensor_storage_size_for_shard) ... ok test_write_item_tensor_storage_size_for_tensor (__main__.TestPlannerAPI.test_write_item_tensor_storage_size_for_tensor) ... ok ---------------------------------------------------------------------- Ran 3 tests in 1.036s OK [root@734774725e04 workspace]# V2.11.0 [root@734774725e04 workspace]# python /workspace/ascend-pytorch-v2.11.0/test/distributed/checkpoint/test_planner_api.py -v test_write_item_tensor_storage_size_for_non_tensor (__main__.TestPlannerAPI.test_write_item_tensor_storage_size_for_non_tensor) ... ok test_write_item_tensor_storage_size_for_shard (__main__.TestPlannerAPI.test_write_item_tensor_storage_size_for_shard) ... ok test_write_item_tensor_storage_size_for_tensor (__main__.TestPlannerAPI.test_write_item_tensor_storage_size_for_tensor) ... ok ---------------------------------------------------------------------- Ran 3 tests in 1.021s OK [root@734774725e04 workspace]# V2.12.0 [root@734774725e04 workspace]# python /workspace/ascend-pytorch-v2.12.0/test/distributed/checkpoint/test_planner_api.py -v test_write_item_tensor_storage_size_for_non_tensor (__main__.TestPlannerAPI.test_write_item_tensor_storage_size_for_non_tensor) ... ok test_write_item_tensor_storage_size_for_shard (__main__.TestPlannerAPI.test_write_item_tensor_storage_size_for_shard) ... ok test_write_item_tensor_storage_size_for_tensor (__main__.TestPlannerAPI.test_write_item_tensor_storage_size_for_tensor) ... ok ---------------------------------------------------------------------- Ran 3 tests in 1.010s OK [root@734774725e04 workspace]# master [root@734774725e04 workspace]# python /workspace/ascend-pytorch-master/test/distributed/checkpoint/test_planner_api.py -v test_write_item_tensor_storage_size_for_non_tensor (__main__.TestPlannerAPI.test_write_item_tensor_storage_size_for_non_tensor) ... ok test_write_item_tensor_storage_size_for_shard (__main__.TestPlannerAPI.test_write_item_tensor_storage_size_for_shard) ... ok test_write_item_tensor_storage_size_for_tensor (__main__.TestPlannerAPI.test_write_item_tensor_storage_size_for_tensor) ... ok ---------------------------------------------------------------------- Ran 3 tests in 1.026s OK [root@734774725e04 workspace]# 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!35401	3 个月前
dynamo	fix: lintrunner --all-files --take NEWLINE -a Co-authored-by: huangjingwei<huangjingwei4@huawei.com> # message auto-generated for no-merge-commit merge: !35875 merge v2.12.0_lintrunner into v2.12.0 fix: lintrunner --all-files --take NEWLINE -a Created-by: huangjingwei Commit-by: huangjingwei Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 1. 执行：`lintrunner --all-files --take NEWLINE -a` 修复换行符，不影响代码逻辑。 2. 屏蔽两个分布式用例，已提issue，给对应接口人跟踪。 # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】不涉及 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!35875	3 个月前
expect	!8437 Add official source code for test_autograd.py Merge pull request !8437 from yuhaiyan/master-dev1	2 年前
functorch	!16093 Update test_common_rules.py/test_eager_transformers.py Merge pull request !16093 from yuhaiyan/master-dev2	1 年前
fx	test(fx): add test for Tracer.getattr Co-authored-by: m0_45651774<2759927519@qq.com> # message auto-generated for no-merge-commit merge: !35500 merge test-fx-Tracer-getattr-v2.12.0 into v2.12.0 test(fx): add test for Tracer.getattr Created-by: m0_45651774 Commit-by: m0_45651774 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260513 --> # 【API功能】 API: `torch.fx.Tracer.getattr` 功能： Tracer.getattr 是 PyTorch FX 符号追踪（Symbolic Tracing）过程中的核心钩子函数之一，它充当了 Python 对象属性访问与 FX 计算图节点之间的桥梁。torch.fx.Tracer.getattr 的主要职责是在追踪（Trace）阶段，拦截对 nn.Module 实例属性的访问操作，并将其转化为计算图（Graph）中的一个 get_attr 节点。在 PyTorch FX 的设计中，Tracer 通过重写 Python 的属性访问机制，确保当代码尝试读取模型的权重（如 self.linear.weight）、偏置或子模块时，不会直接返回底层的 Tensor 或 Module 对象，而是返回一个代表该值的符号代理（Proxy）。 # 【用例完整性】经 PyTorch 社区代码检索，v2.7.1/v2.8.0/v2.9.0/v2.10.0/v2.11.0均无torch.fx.Tracer.getattr 测试用例覆盖，因此需要新增 torch-npu 自有用例。 # 【修改方案】对目标 API 进行现状核查，确认在当前v2.7.1/v2.8.0/v2.9.0/v2.10.0/v2.11.0已实际支持，不需要补齐底层实现，本 PR 仅补充torch.fx.Tracer.getatt的测试用例至test/fx/test_fx_tracer_transformer_api.py。 # 【版本说明】经 PyTorch 社区代码检索，v2.7.1/v2.8.0/v2.9.0/v2.10.0/v2.11.0均存在该API # 【NPU适配分析】 - 职责：在符号追踪过程中，处理模型代码对属性（如 `self.submodule.weight` 或 `self.layers.__len__`）的访问。决定该属性是直接返回原始值（视为常量），还是记录为图中的 `get_attr` 节点。 - 实现层次：该方法属于 `Tracer` 类的 Python 成员函数，其逻辑由 PyTorch 框架在 Python 层完成（例如通过 `super().getattr()` 调用默认行为或用户自定义的 `is_leaf_module` 判断）。 - 设备无关性：`getattr` 操作的对象是 `torch.nn.Module` 或 Python 对象，并不直接接触 `torch.Tensor` 的底层设备内存。即使访问的张量位于 NPU 上，`getattr` 也只是获取该张量的属性元信息（如 `.shape`, `.dtype`）或子模块引用，不涉及 NPU 侧的 kernel 执行或内存传输。 torch.fx.Tracer.getattr属于纯 Python 框架层面的操作，不需要进行 NPU 适配。 # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】 torch.fx.Tracer.getattr测试用例路径：test/fx/test_fx_tracer_transformer_api.py 验证环境 - 对应版本：torch-npu 2.10.0 - 操作系统：openEuler22.03 - CANN 软件版本：8.5.0 - 安装的软件版本：torch、torch-npu 2.10.0 验证结果： ![ScreenShot_20260513-2120.JPG](https://raw.gitcode.com/user-images/assets/7404318/fd832233-98ea-4c3e-a236-21c8f444fd90/ScreenShot_20260513-2120.JPG 'ScreenShot_20260513-2120.JPG') 结果说明：测试用例执行通过。注意：测试结果图中的test_fx_Tracer_getattr.py文件内容已添加到test/fx/test_fx_tracer_transformer_api.py文件。 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!35500	3 个月前
jit	test(jit):add test for ScriptFunction,ScriptFunction.get_debug_state,ScriptFunction.save,and ScriptFunction.save_to_buffer Co-authored-by: fengwu154<1067935138@qq.com> # message auto-generated for no-merge-commit merge: !35415 merge test-torch-jit-ScriptFunction-2.12.0 into v2.12.0 test(jit):add test for ScriptFunction,ScriptFunction.get_debug_state,ScriptFunction.save,and ScriptFunction.save_to_buffer Created-by: fengwu154 Commit-by: fengwu154 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 https://gitcode.com/Ascend/pytorch/issues/1904 - [ ] 需求 - [ ] 问题单 - [x] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 1. `get_debug_state` 适配 - 使用 PyTorch 内置的 `enable_profiling_mode_for_profiling_tests()` 上下文强制启用 profiling executor。 - 执行两次函数（一次 profile，一次 optimize），确保优化计划生成。 - 通过 `GRAPH_EXECUTOR` 常量判断执行器模式，非 profiling 模式自动跳过测试。 2. `save` 适配 - API 签名仅接受字符串路径，因此使用 `tempfile.TemporaryDirectory` 创建临时文件，避免对文件系统的污染。 3. `save_to_buffer` 适配 - 验证返回类型为 `bytes`，并通过 `io.BytesIO` 模拟文件流加载，与保存前输出一致。 - 原测试缺口：`test_jit.py` 虽大量使用 `ScriptFunction`，但缺少对其类型、`save_to_buffer` 及 `save` 直接路径的专门测试；`get_debug_state` 仅在特定测试中偶现调用，未系统验证。 - 新增覆盖： - 类型断言：确保 `@torch.jit.script` 生成的函数为 `torch.jit.ScriptFunction` 实例。 - 调试状态：验证在 profiling 模式下两次执行后可获得有效状态对象。 - 序列化闭环：通过 `save_to_buffer` → `load` 和 `save(file)` → `load` 验证序列化后语义等价。 - 边界适配：`get_debug_state` 通过 `@unittest.skipIf(GRAPH_EXECUTOR != ProfilingMode.PROFILING, ...)` 仅在有 profiling 执行器的环境中运行，避免内部断言错误。 - 新增测试文件 `test/jit/test_jit_ScriptFunction.py`。 - 实现 4 个测试用例，覆盖 `ScriptFunction` 类型、调试状态获取、内存序列化与文件序列化。 - 使用条件跳过（`@unittest.skipIf`）适配不同执行器模式（profiling / simple / legacy）。 - 不依赖外部文件，所有测试可在任意环境运行。 # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】 `root@devserver-fae-mirrors:/home/aicc/wf/api/2.7.1/pytorch/test# python test_jit_ScriptFunction.py .... ---------------------------------------------------------------------- Ran 4 tests in 10.120s OK` 覆盖版本： v2.7.1 v2.9.0 v2.10.0 v2.11.0 v2.12.0 master # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!35415	3 个月前
nn	skip ci failed testcase Co-authored-by: XianglongZeng<zengxianglong1@huawei.com> # message auto-generated for no-merge-commit merge: !36277 merge v2.12.0 into v2.12.0 skip ci failed testcase Created-by: probiotics_53 Commit-by: XianglongZeng Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 https://gitcode.com/Ascend/op-plugin/pull/4958合入受阻，下掉一批与此次修改无关的门禁失败用例： ## 1. `_afd/test_attention_worker_schedule` `File "/workspace/CODE/pytorch_ut/test/_afd/test_attention_worker_schedule.py", line 4, in <module> import torchair ModuleNotFoundError: No module named 'torchair' [ERROR] ERR99999 UNKNOWN applicaiton exception` ## 2. `_afd/test_ffn_worker_schedule` 同上：`ModuleNotFoundError: No module named 'torchair'` + `ERR99999 UNKNOWN` ## 3. `custom_ops/test_npu_multi_head_attention` `File "/workspace/CODE/pytorch_ut/test/custom_ops/test_npu_multi_head_attention.py", line 98 self.assertRtolEqual(cpu_result, npu_result) AssertionError: result error` 精度比较失败（NPU 和 CPU 计算结果差异超阈值）。 ## 4. `distributed/pipelining/test_schedule`（2 个 error） `ERROR [0.288s]: test_grad_with_split_b_w schedule._load_actions(...) AttributeError: '_PipelineScheduleRuntime' object has no attribute '_load_actions' ERROR [0.058s]: test_grad_with_v_schedule torch.distributed.init_process_group(...)` `_PipelineScheduleRuntime` API 缺失 `_load_actions` 方法，distributed 接口不兼容。 ## 5. `nn/test_module_hooks` `File "/workspace/CODE/pytorch_ut/test/nn/test_module_hooks.py", line 1487 with self.assertRaisesRegex(RuntimeError, "Output 0 of BackwardHookFunctionBackward is " AssertionError: expected: "Output 0 of BackwardHookFunctionBackward is a view..." got: "Output 0 of BackwardHookFunction is a view..."` PyTorch 报错信息中类名变更（`BackwardHookFunctionBackward` → `BackwardHookFunction`），正则匹配失败。 ## 6. `test_multiprocessing_api` `File "/workspace/CODE/pytorch_ut/test/test_multiprocessing_api.py", line 215 self.assertTrue(torch.equal(npu_tensor.cpu(), reconstructed_npu.cpu())) AssertionError: False is not true` 多进程 tensor 序列化/反序列化后数据不一致。 ## 7. `test_ops _complex`（3 个 error，同一根因） `RuntimeError: _index_put_impl_:IndexPutKernelNpuOpApi.cpp:82 call aclnnIndexPutImpl failed, error code is 161002 AclNN_Parameter_Error(EZ1001): Tensor self not implemented for DT_COMPLEX64, should be in dtype support list [DT_FLOAT, DT_INT32, DT_INT64, ...]` ACL 底层 `aclnnIndexPutImpl` 不支持 complex64/complex128 数据类型。 ## 总结 \| # \| 测试 \| 根因 \| 与 0d CPU tensor 修改关系 \| \|---\|------\|------\|:--:\| \| 1 \| attention_worker_schedule \| 环境缺 `torchair` 包 \| 无关 \| \| 2 \| ffn_worker_schedule \| 环境缺 `torchair` 包 \| 无关 \| \| 3 \| multi_head_attention \| 精度差异 \| 无关 \| \| 4 \| test_schedule \| distributed API `_load_actions` 缺失 \| 无关 \| \| 5 \| module_hooks \| PyTorch 错误消息类名变更 \| 无关 \| \| 6 \| multiprocessing_api \| 多进程 tensor 序列化不一致 \| 无关 \| \| 7 \| test_ops _complex \| ACL 不支持 complex dtype \| 无关 \| https://gitcode.com/Ascend/op-plugin/pull/4958 涉及的 15 个 `test_base_ops/` 测试全部 success。 # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】不涉及 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!36277	3 个月前
npu	[fix] support memory reuse in launch_host_func_pending Co-authored-by: Chukun Wang<wangchukun@huawei.com> # message auto-generated for no-merge-commit merge: !36081 merge v2.12.0 into v2.12.0 [fix] support memory reuse in launch_host_func_pending Created-by: kita-ikuyo Commit-by: Chukun Wang Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> https://gitcode.com/Ascend/pytorch/issues/2009 - [x] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 > 请描述修改内容的具体实现，涉及哪些组件之间进行交互，可以用1、2、3、...进行罗列\ > 如果是需求或者重构类的PR，需要补充详细设计文档（说明上下游组件关系、时序图、类图、DFX能力等内容） launch_host_func_pending由直接传递Tensor改为传递tensor地址，由callback thread根据地址做h2h拷贝，拷贝结束后原始tensor的pinned memory可以被释放并复用。回调线程/python线程使用被拷贝的tensor值做后续print/save，并在print/save结束后释放内存，以节省运行时host内存使用。 # 【资料变更】 > 请确认是否涉及资料变更。如涉及，需要在PR中体现，并简要说明修改内容。如不涉及，需填写“不涉及” 不涉及 # 【接口变更】 > 请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及，需要详细说明接口以及对应的变更内容，同时需要在资料中体现。如不涉及，需填写“不涉及” 不涉及 # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图验证ok，修改前整网出现host oom，修改后解决。 ![2.jpg](https://raw.gitcode.com/user-images/assets/7404318/0de6a8bb-47bd-4f6c-8c33-869dd98c6df9/2.jpg '2.jpg') # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!36081	3 个月前
onnx	Skip failing torch official TestCase tests and disable unsupported test cases Co-authored-by: yuhaiyan<yuhaiyan8@huawei.com> Co-authored-by: yuhaiyan8<yuhaiyan8@h-partners.com> # message auto-generated for no-merge-commit merge: !32140 merge cherry-pick-mr-32139-1774073503832-auto into master Skip failing torch official TestCase tests and disable unsupported test cases Created-by: yuhaiyan Commit-by: yuhaiyan;yuhaiyan8 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 1. 暂时跳过失败用例 2. 已提单跟进修复 # 【资料变更】 > 请确认是否涉及资料变更。如涉及，需要在PR中体现，并简要说明修改内容。如不涉及，需填写“不涉及” # 【接口变更】 > 请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及，需要详细说明接口以及对应的变更内容，同时需要在资料中体现。如不涉及，需填写“不涉及” # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [ ] 代码注释完备，正确记录错误日志 - [ ] 代码实现进行了返回值、空指针等校验 - [ ] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [ ] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!32140	4 个月前
optim	test: fix failed test cases in test_optim Co-authored-by: 21xiu<caixiuxiu1@huawei.com> # message auto-generated for no-merge-commit merge: !35785 merge cherry-pick-mr-35784-1778835857779-auto into v2.12.0 test: fix failed test cases in test_optim Created-by: xiu_21 Commit-by: 21xiu Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [x] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】修复test_optim失败的测试用例： 1. RMSprop优化器内部实现依赖张量操作，将test_rmsprop(optim.test_optim.TestDifferentiableOptimizer)中的state["step"]=0修改为state["step"]=torch.zeros(()) 2. GPU版本已删除代码块，删除用例test_sgd(optim.test_optim.TestOptim) 3. GPU版本已删除测试，删除用例test_sparse_adam(optim.test_optim.TestOptim) # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】 ![test_rmsprop.png](https://raw.gitcode.com/user-images/assets/7404318/fce7a28a-39be-4bdb-9168-7ab249e1212f/test_rmsprop.png 'test_rmsprop.png') # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!35785	3 个月前
profiler	[torch_2.12.0]修复cann_export中找不到msprof二进制文件日志提示问题 Co-authored-by: yuliangbin<y1416490440@163.com> # message auto-generated for no-merge-commit merge: !35754 merge fix_tid_2.12 into v2.12.0 [torch_2.12.0]修复cann_export中找不到msprof二进制文件日志提示问题 Created-by: yu_liangbin Commit-by: yuliangbin Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单 https://gitcode.com/Ascend/pytorch/issues/1894 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 > 请描述修改内容的具体实现，涉及哪些组件之间进行交互，可以用1、2、3、...进行罗列\ > 如果是需求或者重构类的PR，需要补充详细设计文档（说明上下游组件关系、时序图、类图、DFX能力等内容） 1.使用Profiler解析过程中，如果msprof的二进制文件找不到，当前校验会在os.stat()时抛出异常，如果msprof的二进制文件不存在，后续的权限校验其实无意义，因此权限校验的前提时msprof的二进制文件存在，所以要在权限校验前判断msprof的二进制文件是否存在，存在即校验，反之不校验； 2.cleancode修改 # 【资料变更】 > 请确认是否涉及资料变更。如涉及，需要在PR中体现，并简要说明修改内容。如不涉及，需填写“不涉及” 不涉及 # 【接口变更】 > 请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及，需要详细说明接口以及对应的变更内容，同时需要在资料中体现。如不涉及，需填写“不涉及” 不涉及 # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图 source CANN包，且把msprof的二进制文件删除，预期结果： ![image.png](https://raw.gitcode.com/user-images/assets/7404318/8aa1e8ef-d975-4f6b-ab4f-f3a0c715ff0f/image.png 'image.png') 抛出msprof二进制文件找不到ERROR打屏，而非os.stat的错误 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [ ] 代码注释完备，正确记录错误日志 - [ ] 代码实现进行了返回值、空指针等校验 - [ ] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [ ] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!35754	3 个月前
testing	AI assist developer for python dt master Co-authored-by: aksksks<mengzichao@h-partners.com> # message auto-generated for no-merge-commit merge: !26538 merge master into master AI assist developer for python dt master Created-by: aksksks Commit-by: aksksks 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!26538	9 个月前
trans_contiguous	[feat]TensorTo support preserve_format consistent with GPU Co-authored-by: chenshuai<chenshuai81@huawei.com> # message auto-generated for no-merge-commit merge: !35353 merge cherry-pick-mr-34696-1778548963126-auto into v2.12.0 [feat]TensorTo support preserve_format consistent with GPU Created-by: culechan Commit-by: culechan;chenshuai Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [x] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 1、NPU上tensorto支持preserve_format模式。行为逻辑上对齐pytorch原生框架 2、增加preserve_format模式相关的测试用例 # 【资料变更】不涉及，原生API表格中未体现这部分内容 # 【接口变更】 Tensor.to默认使用且支持preserve_format模式 # 【功能验证】 ![image.png](https://raw.gitcode.com/user-images/assets/7404318/26dfe926-6f3c-4124-9680-486ed681e868/image.png 'image.png') ![image.png](https://raw.gitcode.com/user-images/assets/7404318/a46d3e66-4a4c-4f9f-9531-78b30d665b58/image.png 'image.png') # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!35353	3 个月前
unsupported_test_cases	skip ci failed testcase Co-authored-by: XianglongZeng<zengxianglong1@huawei.com> # message auto-generated for no-merge-commit merge: !36277 merge v2.12.0 into v2.12.0 skip ci failed testcase Created-by: probiotics_53 Commit-by: XianglongZeng Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 https://gitcode.com/Ascend/op-plugin/pull/4958合入受阻，下掉一批与此次修改无关的门禁失败用例： ## 1. `_afd/test_attention_worker_schedule` `File "/workspace/CODE/pytorch_ut/test/_afd/test_attention_worker_schedule.py", line 4, in <module> import torchair ModuleNotFoundError: No module named 'torchair' [ERROR] ERR99999 UNKNOWN applicaiton exception` ## 2. `_afd/test_ffn_worker_schedule` 同上：`ModuleNotFoundError: No module named 'torchair'` + `ERR99999 UNKNOWN` ## 3. `custom_ops/test_npu_multi_head_attention` `File "/workspace/CODE/pytorch_ut/test/custom_ops/test_npu_multi_head_attention.py", line 98 self.assertRtolEqual(cpu_result, npu_result) AssertionError: result error` 精度比较失败（NPU 和 CPU 计算结果差异超阈值）。 ## 4. `distributed/pipelining/test_schedule`（2 个 error） `ERROR [0.288s]: test_grad_with_split_b_w schedule._load_actions(...) AttributeError: '_PipelineScheduleRuntime' object has no attribute '_load_actions' ERROR [0.058s]: test_grad_with_v_schedule torch.distributed.init_process_group(...)` `_PipelineScheduleRuntime` API 缺失 `_load_actions` 方法，distributed 接口不兼容。 ## 5. `nn/test_module_hooks` `File "/workspace/CODE/pytorch_ut/test/nn/test_module_hooks.py", line 1487 with self.assertRaisesRegex(RuntimeError, "Output 0 of BackwardHookFunctionBackward is " AssertionError: expected: "Output 0 of BackwardHookFunctionBackward is a view..." got: "Output 0 of BackwardHookFunction is a view..."` PyTorch 报错信息中类名变更（`BackwardHookFunctionBackward` → `BackwardHookFunction`），正则匹配失败。 ## 6. `test_multiprocessing_api` `File "/workspace/CODE/pytorch_ut/test/test_multiprocessing_api.py", line 215 self.assertTrue(torch.equal(npu_tensor.cpu(), reconstructed_npu.cpu())) AssertionError: False is not true` 多进程 tensor 序列化/反序列化后数据不一致。 ## 7. `test_ops _complex`（3 个 error，同一根因） `RuntimeError: _index_put_impl_:IndexPutKernelNpuOpApi.cpp:82 call aclnnIndexPutImpl failed, error code is 161002 AclNN_Parameter_Error(EZ1001): Tensor self not implemented for DT_COMPLEX64, should be in dtype support list [DT_FLOAT, DT_INT32, DT_INT64, ...]` ACL 底层 `aclnnIndexPutImpl` 不支持 complex64/complex128 数据类型。 ## 总结 \| # \| 测试 \| 根因 \| 与 0d CPU tensor 修改关系 \| \|---\|------\|------\|:--:\| \| 1 \| attention_worker_schedule \| 环境缺 `torchair` 包 \| 无关 \| \| 2 \| ffn_worker_schedule \| 环境缺 `torchair` 包 \| 无关 \| \| 3 \| multi_head_attention \| 精度差异 \| 无关 \| \| 4 \| test_schedule \| distributed API `_load_actions` 缺失 \| 无关 \| \| 5 \| module_hooks \| PyTorch 错误消息类名变更 \| 无关 \| \| 6 \| multiprocessing_api \| 多进程 tensor 序列化不一致 \| 无关 \| \| 7 \| test_ops _complex \| ACL 不支持 complex dtype \| 无关 \| https://gitcode.com/Ascend/op-plugin/pull/4958 涉及的 15 个 `test_base_ops/` 测试全部 success。 # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】不涉及 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!36277	3 个月前
utils	【feat】NPUAffinityController support multiple rangs of affinity cpu core Co-authored-by: zhaoyu<nanzhaogang@qq.com> # message auto-generated for no-merge-commit merge: !35272 merge v2.12.0-Affinity into v2.12.0 【feat】NPUAffinityController support multiple rangs of affinity cpu core Created-by: zhaoyu65 Commit-by: zhaoyu Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [x] 需求 [#1816](https://gitcode.com/Ascend/pytorch/issues/1816) - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 ## 1、修改内容总体本次 commit 的核心目的是让 NPUAffinityController 支持绑定多个不连续的 CPU 核范围。同时修改了lintrunner工具扫描出来的代码规范问题。主要技术变更： - 将数据结构从 `CoreIdRange {start, end}` 改为 `CoreIdList = std::set<CoreId>`，支持离散/多段 CPU 核集合，set容器符合语义，不存在重复值 - 支持解析查出来的多段CPU亲和段 - 对GetAffinityCPUInfo.cpp和NPUAffinityController.cpp重构，优化函数圏复杂度，将cpp内部调用函数使用namespace或static关键词封装， --- ## 2、各文件修改的内容和目的 \| 文件 \| 修改内容 \| 目的 \| \|------\|----------\|------\| \| NPUAffinityController.h \| 将 `CoreIdRange` 结构体替换为 `CoreIdList = std::set<CoreId>`，新增 `SetThreadAffinity(const CoreIdList)` 接口 \| 数据结构重构，支持多核集合 \| \| NPUAffinityController.cpp \| 重构核心绑定逻辑：解析配置支持 `npu0:0-1,npu0:4-9` 多段格式；新增 `formatCoreRange()` 输出函数；重构 `parseDeviceCoreRange()` 解析多段范围 \| 核心实现支持多段 CPU 核绑定 \| \| GetAffinityCPUInfo.cpp \| 将 `parseAffinityCPU()` 改为 `parseAffinityCores()`，支持逗号分隔的多段范围；重构 `GetExclusiveAffinityCPU()` 的核分配逻辑 \| DCMI 接口层适配新数据结构，避免重复初始化 \| \| GetAffinityCPUInfo.h \| 简化接口，对其他cpp文件只保留 `GetAffinityCores(int card_id)`接口 \| 接口统一 \| \| Module.cpp \| Python 绑定增加对 `PyList` 参数的解析，支持python内部接口侧传入 `[0, 1, 4, 5]` 格式的核列表 \| Python API 支持新参数类型 \| \| affinity.py \| `_set_thread_affinity()` 支持 `List[int]` 或 `List[List[int]]` 参数，如 `[[0,5], [8,10]]` \| Python 层 API 扩展 \| \| torch_npu_schema.json \| 更新签名 `Union[List[int], List[List[int]], NoneType]` \| 类型提示更新 \| \| test_affinity.py \| 新增加多段CPU核相关测试用例 \| 新增用例 \| --- ## 3、修改后的影响 ### 功能增强 - 用户可配置多段 CPU 核范围：如 `CPU_AFFINITY_CONF='1,npu0:0-3,npu0:8-15'`，同一设备可绑定不连续的核，取并集 - API 更灵活：`set_thread_affinity()` 新增接受 `[[0,5], [8,10]]` 格式参数的功能 ### 行为变化 - 环境变量 `CPU_AFFINITY_CONF` 同一设备多次指定核范围会合并（如 `npu0:0-1,npu0:4-9` 绑定核 0,1,4,5,6,7,8,9） ### 代码改进 - 新增 `formatCoreRange()` 函数统一输出格式（如 `0-3,8-15`） - 代码结构更清晰：解析函数拆分为 `parseNpuAffineMode()`、`parseLazyBindMode()`、`parseForceMode()`、`parseDeviceCoreRange()` --- # 【资料变更】在后续2.7.1分支的PR中修改 # 【接口变更】 CPU_AFFINITY_CONF环境变量和set_thread_affinity()接口变更方案已通过SEG例会评审，兼容旧版本； # 【功能验证】自测OK ![image.png](https://raw.gitcode.com/user-images/assets/7404318/d1efc099-e0c4-49e1-87ba-396be42fc56c/image.png 'image.png') 验证环境变量CPU_AFFINITY_CONF bash #!/bin/bash # 设置环境变量 export ASCEND_PROCESS_LOG_PATH=$(pwd) export ASCEND_GLOBAL_LOG_LEVEL=0 echo -e 'import torch\na=torch.tensor([1],device="npu")\nb=a+1\n' > add.py # 定义配置和期望值数组 configs=( "1,npu_affine:1,npu0:0-5" "1,npu_affine:1,npu0:0-5,npu1:6-7,npu2:8-8,npu2:10-11,npu0:12-19" "1,npu_affine:1,npu0:0-5,npu1:10-13,npu0:4-7" ) values=( "0-5" "0-5,12-19" "0-7" ) run_test() { local index=$1 local conf=$2 local expected_val=$3 export CPU_AFFINITY_CONF="$conf" # 清理旧日志，确保获取的是本次运行的日志 rm -rf debug/ # 执行测试脚本 python add.py # 检查 python 执行是否成功，可选 if [ $? -ne 0 ]; then echo "[ERROR] Index $index: python add.py failed." >> result.log return 1 fi # 提取 affinity 值 # 使用 find 避免 glob 匹配不到文件时的报错，或者保留原来的 glob 但确保目录存在 local log_file log_file=$(find debug/plog -name ".log" 2>/dev/null \| head -1) if [ -z "$log_file" ]; then echo "[ERROR] Index $index: No log file found in debug/plog/" >> result.log return 1 fi local affinity_line affinity_line=$(grep "Device 0 set acl_thread affinity to" "$log_file" 2>/dev/null \| head -1) if [ -z "$affinity_line" ]; then echo "[ERROR] Index $index: Affinity line not found in log." >> result.log return 1 fi # 提取 "affinity to" 后面的具体值 local affinity_value # sed 提取最后一个非空字段，或者根据具体日志格式调整 affinity_value=$(echo "$affinity_line" \| sed -n 's/.affinity to $[^ ]$./\1/p') # 去除可能的首尾空白字符 (trim) affinity_value=$(echo "$affinity_value" \| xargs) # 比较实际值与期望值 if [ "$affinity_value" == "$expected_val" ]; then echo "测试环境变量 CPU_AFFINITY_CONF $conf: 的结果为：SUCCESS (Actual: ${affinity_value} == Expected: ${expected_val})" else echo "测试环境变量 CPU_AFFINITY_CONF $conf: 的结果为：FAIL (Actual: ${affinity_value} != Expected: ${expected_val})" fi } # 循环遍历配置 # 使用 C 风格的 for 循环以便同时获取索引 for ((i=0; i<${#configs[@]}; i++)); do run_test "$i" "${configs[$i]}" "${values[$i]}" done echo "All tests completed." 测试结果为：所有用例都通过 ![image.png](https://raw.gitcode.com/user-images/assets/7404318/b09a86c8-ce2d-4e50-a1c0-fc8df43e65f9/image.png 'image.png') # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!35272	3 个月前
README.md	!21966 Document description optimization Merge pull request !21966 from shaoyf/master_readme	1 年前
adapt_testcases_to_npu.py	!20432 Fixed the failed testcases Merge pull request !20432 from yuhaiyan/master-dev1	1 年前
adaptive_tests.txt	!20432 Fixed the failed testcases Merge pull request !20432 from yuhaiyan/master-dev1	1 年前
allowlist_for_publicAPI.json	TORCH MAIN SYNC: switch to release candidate of torch 2.12 update Co-authored-by: chansinging<chenxingying2@huawei.com> Co-authored-by: Chansinging<1183520536@qq.com> # message auto-generated for no-merge-commit merge: !35103 merge main_sync_0509 into master TORCH MAIN SYNC: switch to release candidate of torch 2.12 update Created-by: chansinging Commit-by: Chansinging;chansinging Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】同步社区演进，例行更新 lintrunner规避，见[issue](https://gitcode.com/Ascend/pytorch/issues/1865) # 【修改方案】出包参考社区改动，https://github.com/pytorch/pytorch/pull/177570/ # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】本地A3服务器和ci流水线已跑通测试 ![image.png](https://raw.gitcode.com/user-images/assets/7404318/a7797f8a-f0ad-4dc4-860a-3c1ff2d11f53/image.png 'image.png') # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!35103	3 个月前
deprecated_apis.json	!19783 Remove expose for deprecated apis. Merge pull request !19783 from 刘嘉巍/master	1 年前
get_failed_ut_from_log.py	fix: lintrunner --all-files --take NEWLINE -a Co-authored-by: huangjingwei<huangjingwei4@huawei.com> # message auto-generated for no-merge-commit merge: !35875 merge v2.12.0_lintrunner into v2.12.0 fix: lintrunner --all-files --take NEWLINE -a Created-by: huangjingwei Commit-by: huangjingwei Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 1. 执行：`lintrunner --all-files --take NEWLINE -a` 修复换行符，不影响代码逻辑。 2. 屏蔽两个分布式用例，已提issue，给对应接口人跟踪。 # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】不涉及 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!35875	3 个月前
get_synchronized_files.sh	!11238 To get synchronized files. Merge pull request !11238 from yuhaiyan/master-dev1	2 年前
requirements.txt	Release 2.11.0rc1 Co-authored-by: Chansinging<1183520536@qq.com> # message auto-generated for no-merge-commit merge: !35503 merge v2.12.0rc1 into v2.12.0 Release 2.11.0rc1 Created-by: chansinging Commit-by: Chansinging Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单 [版本升级](https://gitcode.com/Ascend/pytorch/issues/1941) - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】社区演进和版本升级 # 【资料变更】修改内容参考[2.11rc合入](https://gitcode.com/Ascend/pytorch/pull/32267)，做2.12版本升级 # 【接口变更】不涉及 # 【功能验证】 ![image.png](https://raw.gitcode.com/user-images/assets/7404318/a2a92a81-1b46-4da4-8daf-0542aa7d3d7e/image.png 'image.png') ![image.png](https://raw.gitcode.com/user-images/assets/7404318/8efb16e3-9769-4bdd-aaf6-5c25f391ade6/image.png 'image.png') # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!35503	3 个月前
run_ops_tests.py	!6717 [Codecheck] test codecheck. Merge pull request !6717 from will-devil/myccheck	2 年前
run_test.py	!12854 [RFC][Feature] Add support for device extension autoloading Merge pull request !12854 from Yuanhao Ji/feat/autoload	1 年前
sample_torch_npu_run_store.py	!13891 add ParallelTcpStore and register to rendezvous Merge pull request !13891 from 叶子凡/master_TCP	2 年前
test_accelerator.py	add test case for event elapsed_time Co-authored-by: chenkun<chenkun82@huawei.com> # message auto-generated for no-merge-commit merge: !26731 merge master_event_ut into master add test case for event elapsed_time Created-by: kuhn7 Commit-by: chenkun 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!26731	9 个月前
test_autocast.py	feat(autocast): 支持float32作为自动混合精度计算的数据类型 Co-authored-by: leland17<lileliao@foxmail.com> # message auto-generated for no-merge-commit merge: !30512 merge master into master feat(autocast): 支持float32作为自动混合精度计算的数据类型 Created-by: leland17 Commit-by: leland17 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > (如有)请关联需求文档/issue链接 issue：[支持float32作为自动混合精度计算的数据类型](https://gitcode.com/Ascend/pytorch/issues/1537) - [x] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】当前现状with autocast(device="npu", dtype=torch.float32)暂时不支持按照float32的精度获取结果。同liwei有过详细的讨论方案，并且确定了如下的支持思路在autocast_utils.py中支持float32的数据类型支持。 # 【资料变更】 > 不涉及 # 【接口变更】 > 不涉及 # 【功能验证】 `bash cd ${pytorch}/test && python -m unittest test_autocast.TestAutocastNPUfp32` 新增4个用例场景，验证暂无问题 ![image.png](https://raw.gitcode.com/user-images/assets/7404318/ae524921-b8cd-4790-b2c2-6ef636ca7229/image.png 'image.png') 整体test_autocast.py的用例执行结果 ![image.png](https://raw.gitcode.com/user-images/assets/7404318/c4ba297d-8d9d-446a-93d7-f6031525dde3/image.png 'image.png') # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!30512	6 个月前
test_autograd.py	!15327 Delete some skipped test cases for test_convolution.py / test_autograd.py / test_optim.py. Merge pull request !15327 from yuhaiyan/master-dev5	1 年前
test_autoload.py	!12854 [RFC][Feature] Add support for device extension autoloading Merge pull request !12854 from Yuanhao Ji/feat/autoload	1 年前
test_binary_ufuncs.py	!12238 Update test_binary_ufuncs of pt 2.4 Merge pull request !12238 from 张向龙3/binary_ufuncs_24	2 年前
test_cat.py	add_test Co-authored-by: sunhaolun<sunhaolun3@huawei.com> # message auto-generated for no-merge-commit merge: !33714 merge master into master add_test Created-by: sunhaolun Commit-by: sunhaolun Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [x] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】为以下API增加用例功能 Tensor.copy_ Tensor.narrow torch.cat torch.chunk torch._chunk_cat # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】新增文件 test_cat.py test_chunk.py test_chunk_cat.py test_copy_.py test_narrow.py # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!33714	4 个月前
test_chunk.py	add_test Co-authored-by: sunhaolun<sunhaolun3@huawei.com> # message auto-generated for no-merge-commit merge: !33714 merge master into master add_test Created-by: sunhaolun Commit-by: sunhaolun Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [x] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】为以下API增加用例功能 Tensor.copy_ Tensor.narrow torch.cat torch.chunk torch._chunk_cat # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】新增文件 test_cat.py test_chunk.py test_chunk_cat.py test_copy_.py test_narrow.py # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!33714	4 个月前
test_chunk_cat.py	add_test Co-authored-by: sunhaolun<sunhaolun3@huawei.com> # message auto-generated for no-merge-commit merge: !33714 merge master into master add_test Created-by: sunhaolun Commit-by: sunhaolun Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [x] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】为以下API增加用例功能 Tensor.copy_ Tensor.narrow torch.cat torch.chunk torch._chunk_cat # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】新增文件 test_cat.py test_chunk.py test_chunk_cat.py test_copy_.py test_narrow.py # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!33714	4 个月前
test_copy_.py	add_test Co-authored-by: sunhaolun<sunhaolun3@huawei.com> # message auto-generated for no-merge-commit merge: !33714 merge master into master add_test Created-by: sunhaolun Commit-by: sunhaolun Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [x] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】为以下API增加用例功能 Tensor.copy_ Tensor.narrow torch.cat torch.chunk torch._chunk_cat # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】新增文件 test_cat.py test_chunk.py test_chunk_cat.py test_copy_.py test_narrow.py # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!33714	4 个月前
test_dataloader.py	!8912 Synchronize the different test cases for v2.2.0. Merge pull request !8912 from yuhaiyan/master-dev1	2 年前
test_dispatch.py	!9500 The test cases support NPU. Merge pull request !9500 from yuhaiyan/master-dev4	2 年前
test_fake_tensor.py	fix: lintrunner --all-files --take NEWLINE -a Co-authored-by: huangjingwei<huangjingwei4@huawei.com> # message auto-generated for no-merge-commit merge: !35875 merge v2.12.0_lintrunner into v2.12.0 fix: lintrunner --all-files --take NEWLINE -a Created-by: huangjingwei Commit-by: huangjingwei Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 1. 执行：`lintrunner --all-files --take NEWLINE -a` 修复换行符，不影响代码逻辑。 2. 屏蔽两个分布式用例，已提issue，给对应接口人跟踪。 # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】不涉及 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!35875	3 个月前
test_flop_counter.py	!10958 Add test files: test_flop_counter/test_jit_autocast/test_jit_fuser/... Merge pull request !10958 from yuhaiyan/master-dev3	2 年前
test_functional_optim.py	!8912 Synchronize the different test cases for v2.2.0. Merge pull request !8912 from yuhaiyan/master-dev1	2 年前
test_functionalization.py	!9365 The test files support NPU. Merge pull request !9365 from 王夏夏/master	2 年前
test_functionalization_of_rng_ops.py	!9365 The test files support NPU. Merge pull request !9365 from 王夏夏/master	2 年前
test_fx.py	!10598 Delete the URL in code. Merge pull request !10598 from yuhaiyan/master-dev7	2 年前
test_fx_experimental.py	!9365 The test files support NPU. Merge pull request !9365 from 王夏夏/master	2 年前
test_fx_passes.py	!9365 The test files support NPU. Merge pull request !9365 from 王夏夏/master	2 年前
test_hub_api.py	test(hub): add test cases for torch.hub.help and torch.hub._get_torch_home APIs Co-authored-by: xiemingda1002<xiemingda1002@163.com> # message auto-generated for no-merge-commit merge: !34071 merge add-test-hub-help-api-master into master test(hub): add test cases for torch.hub.help and torch.hub._get_torch_home APIs Created-by: xiemingda1002 Commit-by: xiemingda1002 Merged-by: ascend-robot Description: # 【合入来源】 > <font color="red">https://gitcode.com/Ascend/pytorch/issues/1703</font>\ - [ ] 需求 - [ ] 问题单 - [x] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】一、API 功能说明 torch.hub.help 和 torch.hub._get_torch_home 是 PyTorch hub 模块中用于文档查询和路径管理的辅助 API，功能说明如下： 1. torch.hub.help(github, model, force_reload=False, skip_validation=False, trust_repo=None) - 文档获取核心能力：接收 GitHub 仓库地址和入口点名称，返回该入口点的文档字符串（docstring），无需实际加载模型或执行代码。 - 远程仓库解析核心能力：支持 "owner/repo[:branch]" 格式的仓库地址，自动解析并获取 hubconf.py 中定义的入口点函数的文档。 - 安全机制支持：通过 skip_validation 参数可跳过 GitHub API 验证（避免 rate limit），trust_repo 参数控制仓库信任策略（False/True/check/None）。 - 无状态调用：多次调用 help 不会产生副作用，不修改缓存状态，不影响后续 load 等操作，不下载仓库代码。 - 返回值灵活性：返回字符串类型文档或 None（若入口点函数未定义 docstring），支持用户预览模型使用说明。 - 典型用途：在加载模型前查看入口点的参数说明、使用示例、依赖信息等。 2. torch.hub._get_torch_home() - 路径解析核心能力：根据环境变量 TORCH_HOME 或默认路径 ~/.cache/torch 确定 torch home 目录。 - 环境变量优先级机制：优先读取 TORCH_HOME 环境变量，若未设置则依次检查 XDG_CACHE_HOME/torch、~/.cache/torch。 - 内部 API 特性：作为 get_dir() 的底层实现，不直接暴露给用户，但可通过 hub 模块访问（用于测试或调试）。 - 路径标准化处理：返回的路径经过 os.path.expanduser 处理，支持 ~ 符号展开，确保路径可用性。 - 返回值：字符串类型路径，非空且为绝对路径。上述 API 均为纯 Python 层实现，不涉及张量计算，仅负责文档查询和路径解析。二、测试文件 test/test_hub_api.py 完整验证该 API 的原因本测试文件针对 PyTorch 社区原生 test/test_hub.py 覆盖不足的 torch.hub.help / _get_torch_home API 进行补充验证，确保昇腾 NPU 环境下功能完整可用： 1. torch.hub.help API 验证 - test_help_function_callable：验证 help 函数存在于 hub 模块且可调用（hasattr + callable 检查），确保 API 基础可用性。 - test_help_returns_none_without_docstring：严格验证 help 函数在入口点无 docstring 时返回 None。使用 mock 本地仓库进行测试，无需外部网络依赖。 - test_help_returns_docstring_with_content：严格验证 help 函数在入口点有 docstring 时返回有效的字符串内容。使用 mock 本地仓库进行测试，检查返回值类型为 str、非空、且包含预期关键字 "EfficientNet"。 2. torch.hub._get_torch_home API 验证 - test_get_torch_home_returns_path：验证 _get_torch_home 返回有效的路径字符串（assertIsInstance(str) + assertGreater(len, 0)），确保路径非空且类型正确。 - test_get_torch_home_with_env_variable：验证 TORCH_HOME 环境变量能正确影响返回路径，使用 tempfile.TemporaryDirectory 创建临时目录作为测试路径，覆盖自定义路径场景。测试后恢复原始环境变量状态（os.environ.pop 或 restore），避免污染其他测试。 3. 测试隔离性保障 - setUp 机制：TestHubHelp 类在 setUp 中创建 mock 本地仓库目录，写入 hubconf.py 文件，包含有/无 docstring 的两个入口点函数。 - tearDown 机制：每个测试后清理临时目录，确保测试用例间无状态污染。 - 独立性设计：两个测试类（TestHubHelp、TestHubGetTorchHome）独立，无共享状态。 - 零外部依赖：使用 unittest.mock.patch 模拟 _get_cache_or_reload，避免对 GitHub 外部仓库的依赖，测试更快、更可靠，适合 CI 环境。测试完整性所有用例覆盖 API 的基础功能验证、接口存在性检查、环境变量交互、测试隔离性保障，完整覆盖昇腾 NPU 适配所需的最小功能集合。三、昇腾 NPU 适配说明（解释为什么无需 NPU 适配） torch.hub.help 和 torch.hub._get_torch_home 均具备硬件无关性、纯 Python 层实现、无底层算子依赖三大核心特征，决定了其无需针对昇腾 NPU 做适配，具体分析： 1. 纯 Python 层抽象，无硬件相关逻辑 - help API 硬件无关性：仅负责解析远程仓库的 hubconf.py 并提取文档字符串，是「文档查询」的纯 Python 抽象接口，不涉及任何硬件相关的计算、存储、通信逻辑。 - _get_torch_home API 硬件无关性：仅负责读取环境变量和构建路径字符串，依赖 os.path 模块进行路径处理，无任何硬件相关的逻辑或依赖。 - 设备无关设计：两个 API 的核心逻辑与 CPU/GPU/NPU 等硬件架构完全解耦，在任意硬件环境下行为一致。 2. 无底层算子 / 内核依赖 - 标准库依赖：两个 API 内部仅调用 Python 标准库（os 路径模块、tempfile 临时文件模块、urllib 网络模块）和 PyTorch 纯 Python 层的 hub 模块。 - 无硬件扩展依赖：未依赖 CUDA/NPU 等硬件相关的扩展库、内核函数或底层驱动。 - help 的远程解析机制：基于 GitHub URL 和 Python import 机制，与硬件架构无关。 - _get_torch_home 的路径处理机制：基于 os.path.expanduser 和 os.getenv，纯文件系统操作。 3. 核心逻辑与硬件解耦 - 文档获取解耦：help API 返回的文档字符串是纯文本信息，与硬件架构无关 —— 无论是 CPU/GPU/NPU 环境，文档查询逻辑完全一致。 - 路径解析解耦：_get_torch_home 返回的路径是通用文件系统路径，不包含任何硬件相关字段（如设备 ID、内存地址），昇腾 NPU 环境下可直接复用。 - 无张量操作：两个 API 均不涉及张量的创建、计算或迁移，测试用例中无 tensor 相关代码，无需 NPU 设备支持。结论 torch.hub.help 和 torch.hub._get_torch_home 均是「硬件无关的纯 Python 层抽象」，核心逻辑不耦合任何特定硬件。测试用例无需导入 torch_npu，API 本身无需适配修改，可直接复用。 # 【资料变更】刷新2.7.1版本下有多个版本资料。 1、torch.hub._get_torch_home API该API昇腾支持，但是在文档中并无描述，因此需要增加。文档修改PR链接：https://gitcode.com/Ascend/pytorch/pull/34365 # 【接口变更】不涉及 # 【功能验证】测试文件路径：test/test_hub_api.py 本地验证结果： ![image.png](https://raw.gitcode.com/user-images/assets/7404318/78941546-13bc-4c7e-9cc7-c6544df41397/image.png 'image.png') 所有测试用例执行通过。torch.hub.help 和 torch.hub._get_torch_home API 核心功能验证正常。两个 API 均为纯 Python 层实现，不涉及张量操作，无需 NPU 设备适配。测试用例无需导入 torch_npu，API 本身可在任意硬件环境稳定运行。验证结果表明，torch.hub.help 和 torch.hub._get_torch_home 在昇腾 NPU 环境下功能正常可用，无需修改源码。 # 【CheckList】 - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签：test(hub) - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!34071	4 个月前
test_import_stats.py	!7605 Add UT files Merge pull request !7605 from yuhaiyan/master-dev1	2 年前
test_indexing.py	fix: skip test_index_put_accumulate_large_tensor Co-authored-by: zhanw_coding<zhanwang@huawei.com> # message auto-generated for no-merge-commit merge: !31414 merge master into master fix: skip test_index_put_accumulate_large_tensor Created-by: zhanw_coding Commit-by: zhanw_coding Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】门禁A1机器cann8.5.0 AsStrided124 算子报错 # 【资料变更】 > 请确认是否涉及资料变更。如涉及，需要在PR中体现，并简要说明修改内容。如不涉及，需填写“不涉及” # 【接口变更】 > 请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及，需要详细说明接口以及对应的变更内容，同时需要在资料中体现。如不涉及，需填写“不涉及” # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!31414	5 个月前
test_jit.py	fix(nn): fix test for jit api: torch.jit.script、torch.jit.trace、torch.jit.save、torch.jit.load Co-authored-by: dinglaiping<1016581171@qq.com> # message auto-generated for no-merge-commit merge: !32402 merge fixtest-jit-api-master into master fix(jit): fix test for jit api: torch.jit.script、torch.jit.trace、torch.jit.save、torch.jit.load Created-by: dinglaiping Commit-by: dinglaiping Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">https://gitcode.com/Ascend/pytorch/issues/1597</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] issue/工单 # 【修改方案】 > 请描述修改内容的具体实现，涉及哪些组件之间进行交互，可以用1、2、3、...进行罗列\ > 如果是需求或者重构类的PR，需要补充详细设计文档（说明上下游组件关系、时序图、类图、DFX能力等内容） 2.7.1及以上版本提示additional_module_tests、new_module_tests和module_tests找不到，要把additional_module_tests换成get_all_nn_module_tests并且删除new_module_tests和module_tests，最后在调用的时候使用get_all_nn_module_tests()。 # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图整体结论: 本次验证基于torch-npu 官方test_jit.py用例全量复用的方式，在完成最小化的文件、代码、环境适配后，于 NPU环境下完成torch.jit.script、torch.jit.trace、torch.jit.save、torch.jit.load四个核心 API 的全量验证，所有指定的测试方法均执行成功。验证结果表明，torch-npu 官方 JIT 用例可完整覆盖四个 API 的核心功能，使用官网用例完成 NPU 适配验证已足够；四个 API 在 NPU环境下的基础能力、协同逻辑均正常可用，完全满足 NPU 场景下模型编译优化、序列化部署、跨设备迁移、推理执行的实际业务需求。 1. torch.jit.script：官方用例中函数重复脚本化操作正常，脚本化模型的序列化 / 反序列化无数据丢失，NPU 设备下编译后模型的参数、设备属性与原模型保持一致，编译优化逻辑正常生效； 2. torch.jit.trace：官方用例中追踪化过程可完整保留模型的 train/eval 状态，字典类型可正常作为示例输入完成追踪，NPU 设备下追踪化模型的前向传播执行正常，输出结果与原模型一致； 3. torch.jit.save：官方用例中模型可成功保存至 zip 压缩包格式，保存过程中可完整保留 NPU 设备信息与模型全状态（参数、缓冲区、计算图），序列化过程无数据遗漏或损坏； 4. torch.jit.load：官方用例中可成功还原模型保存时的 NPU 设备信息，模型中自定义方法可正常导入并调用，zip 压缩包格式的模型可正确解析，加载后模型可直接在 NPU 上推理，执行结果与保存前模型一致。运行日志： root@hostname-fqv42:~/torchnpuapi/torch-npu/2.7.1/pytorch/test# python -m unittest test_jit.TestJit.test_restore_device test_jit.TestJit.test_import_method test_jit.TestJit.test_torch_load_zipfile_check test_jit.TestScript.test_repeated_script_on_function test_jit.TestJit.test_script_fn_pkl test_jit.TestJit.test_trace_retains_train test_jit.TestFrontend.test_dictionary_as_example_inputs_for_jit_trace -v monkeytype is not installed. Skipping tests for Profile-Directed Typing test_restore_device (test_jit.TestJit.test_restore_device) ... ok test_import_method (test_jit.TestJit.test_import_method) ... ok test_torch_load_zipfile_check (test_jit.TestJit.test_torch_load_zipfile_check) ... ok test_repeated_script_on_function (test_jit.TestScript.test_repeated_script_on_function) ... ok test_script_fn_pkl (test_jit.TestJit.test_script_fn_pkl) ... ok test_trace_retains_train (test_jit.TestJit.test_trace_retains_train) ... ok test_dictionary_as_example_inputs_for_jit_trace (test_jit.TestFrontend.test_dictionary_as_example_inputs_for_jit_trace) ... ok ---------------------------------------------------------------------- Ran 7 tests in 0.104s OK root@hostname-fqv42:~/torchnpuapi/torch-npu/2.7.1/pytorch/test# # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!32402	5 个月前
test_jit_api.py	[feat] Add test cases for verifying torch.jit.ScriptModule hook API on NPU Co-authored-by: hxm_<ha_m@foxmail.com> # message auto-generated for no-merge-commit merge: !35723 merge v2.12.0 into v2.12.0 [feat] Add test cases for verifying torch.jit.ScriptModule hook API on NPU Created-by: hxm_ Commit-by: hxm_ Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 issue连接：https://gitcode.com/Ascend/pytorch/issues/1723 - [ ] 需求 - [ ] 问题单 - [x] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 ## 一、背景说明 1. 通过`torch.jit.script()`创建的模型是`RecursiveScriptModule`实例，与测试目的不符，所以本次测试模型均通过继承`torch.jit.ScriptModule`创建。 2. `torch.jit.ScriptModule` 继承自 `torch.nn.Module`，在 `__init__` 执行完毕后通过 `torch.jit.ScriptModule .init_then_script()` 将模块编译为 TorchScript，内部持有 `torch.jit.ScriptModule ._actual_script_module`（`RecursiveScriptModule` 实例）作为 C++ 侧代理。编译过程只迁移参数/子模块/缓冲区的存储，不替换类上的方法，也不影响 hook 相关属性（`torch.jit.ScriptModule ._backward_hooks`、`torch.jit.ScriptModule ._backward_pre_hooks` 等），因此大部分 hook API 在 ScriptModule 子类上行为与普通 Module 一致。 3. `torch.jit.ScriptModule .register_full_backward_hook` 例外的原因：该方法内部会设置 `self._is_full_backward_hook = True`，该属性赋值被 `ScriptModule.__setattr__` 代理到 C++ 侧后，因类型不匹配（C++ 侧期望 NoneType）导致 RuntimeError。这是pytorch自身的bug。 ## 二、API 功能说明 ### 1. torch.jit.ScriptModule.register_full_backward_hook(hook, prepend=False) -> RemovableHandle 在模块上注册反向传播后置 hook。hook 签名为 `hook(module, grad_input, grad_output) -> tuple[Tensor] or None`，在模块梯度计算完成时被调用，可返回新的 grad_input 替代原有值。prepend=True 时 hook 在已有 hook 之前执行。返回 RemovableHandle 用于移除 hook。 ### 2. torch.jit.ScriptModule.register_full_backward_pre_hook(hook, prepend=False) -> RemovableHandle 在模块上注册反向传播前置 hook。hook 签名为 `hook(module, grad_output) -> tuple[Tensor] or None`，在模块梯度计算之前被调用，可返回新的 grad_output 影响后续梯度计算。prepend=True 时 hook 在已有 hook 之前执行。返回 RemovableHandle 用于移除 hook。 ### 3. torch.jit.ScriptModule.register_load_state_dict_pre_hook(hook) -> RemovableHandle 在 load_state_dict 调用前触发 hook。hook 签名为 `hook(module, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs) -> None`，可用于在加载前对 state_dict 进行预处理。返回 RemovableHandle 用于移除 hook。 ### 4. torch.jit.ScriptModule.register_load_state_dict_post_hook(hook) -> RemovableHandle 在 load_state_dict 调用后触发 hook。hook 签名为 `hook(module, incompatible_keys) -> None`，incompatible_keys 包含 missing_keys 和 unexpected_keys，可原地修改。返回 RemovableHandle 用于移除 hook。 ### 5. torch.jit.ScriptModule.register_state_dict_pre_hook(hook) -> RemovableHandle 在 state_dict 调用前触发 hook。hook 签名为 `hook(module, prefix, keep_vars) -> None`，可用于在序列化前执行预处理。返回 RemovableHandle 用于移除 hook。 ### 6. torch.jit.ScriptModule.register_state_dict_post_hook(hook) -> RemovableHandle 在 state_dict 调用后触发 hook。hook 签名为 `hook(module, state_dict, prefix, local_metadata) -> None`，可原地修改 state_dict。返回 RemovableHandle 用于移除 hook。 ## 三、测试用例说明 ### 1. torch.jit.ScriptModule.register_full_backward_hook \| 用例 \| 说明 \| \|------\|------\| \| test_register_full_backward_hook_raises \| 验证调用 register_full_backward_hook 时抛出 RuntimeError \| ### 2. torch.jit.ScriptModule.register_full_backward_pre_hook \| 用例 \| 说明 \| \|------\|------\| \| test_register_full_backward_pre_hook_called \| 验证 hook 在反向传播时被调用 \| \| test_register_full_backward_pre_hook_modify_grad \| 验证 hook 返回修改后的 grad_output 可影响梯度计算（返回全零梯度，输入梯度为零） \| \| test_register_full_backward_pre_hook_prepend \| 验证 prepend=True 时 hook 在已有 hook 之前执行 \| \| test_register_full_backward_pre_hook_remove \| 验证 handle.remove() 后 hook 不再触发 \| ### 3. torch.jit.ScriptModule.register_load_state_dict_pre_hook & torch.jit.ScriptModule.register_load_state_dict_post_hook \| 用例 \| 说明 \| \|------\|------\| \| test_load_state_dict_pre_hook_fires_before_module_and_post_hook \| 验证完整时序 pre_hook -> module(load_state_dict) -> post_hook，post_hook 先注册证明调用顺序与注册顺序无关；pre_hook 修改 state_dict 为全零，post_hook 检查权重已加载为零，证明 module 在 pre 和 post 之间执行 \| \| test_register_load_state_dict_pre_hook_called \| 验证 pre_hook 在 load_state_dict 时被调用，接收到正确的 prefix 参数 \| \| test_register_load_state_dict_pre_hook_with_module \| 验证 pre_hook 接收到的 module 参数就是当前模型实例 \| \| test_register_load_state_dict_pre_hook_remove \| 验证 handle.remove() 后 pre_hook 不再触发 \| \| test_register_load_state_dict_post_hook_called \| 验证 post_hook 在 load_state_dict 后被调用 \| \| test_register_load_state_dict_post_hook_with_module \| 验证 post_hook 接收到的 module 参数就是当前模型实例 \| \| test_register_load_state_dict_post_hook_remove \| 验证 handle.remove() 后 post_hook 不再触发 \| ### 4. torch.jit.ScriptModule.register_state_dict_pre_hook & torch.jit.ScriptModule.register_state_dict_post_hook \| 用例 \| 说明 \| \|------\|------\| \| test_state_dict_pre_hook_fires_before_module_and_post_hook \| 验证完整时序 pre_hook -> module(state_dict) -> post_hook，post_hook 先注册证明调用顺序与注册顺序无关；post_hook 检查 state_dict 已包含模型参数，证明 module 在 pre 和 post 之间执行 \| \| test_register_state_dict_pre_hook_called \| 验证 pre_hook 在 state_dict 时被调用，接收到正确的 prefix 参数 \| \| test_register_state_dict_pre_hook_with_module \| 验证 pre_hook 接收到的 module 参数就是当前模型实例 \| \| test_register_state_dict_pre_hook_remove \| 验证 handle.remove() 后 pre_hook 不再触发 \| \| test_register_state_dict_post_hook_called \| 验证 post_hook 在 state_dict 后被调用，接收到正确的 prefix 参数 \| \| test_register_state_dict_post_hook_with_module \| 验证 post_hook 接收到的 module 参数就是当前模型实例 \| \| test_register_state_dict_post_hook_remove \| 验证 handle.remove() 后 post_hook 不再触发 \| # 【资料变更】 torch.jit.ScriptModule.register_load_state_dict_pre_hook、torch.jit.ScriptModule.register_state_dict_post_hook API昇腾支持，但是在文档中并无描述，因此需要增加。其他API昇腾支持，且在文档中有描述，无需添加。文档修改PR链接：https://gitcode.com/Ascend/pytorch/pull/35840 # 【接口变更】不涉及 # 【功能验证】 \| 接口 \| 状态 \| 说明 \| \|------\|------\|------\| \| `torch.jit.ScriptModule.register_full_backward_hook` \| 不可用 \| 调用时抛出 RuntimeError \| \| `torch.jit.ScriptModule.register_full_backward_pre_hook` \| 可用 \| 正常注册并触发，支持修改 grad_output、prepend 顺序控制、handle 移除 \| \| `torch.jit.ScriptModule.register_load_state_dict_pre_hook` \| 可用 \| 在 load_state_dict 前触发，可接收 module 参数，支持 handle 移除 \| \| `torch.jit.ScriptModule.register_load_state_dict_post_hook` \| 可用 \| 在 load_state_dict 后触发，可接收 module 参数，支持 handle 移除 \| \| `torch.jit.ScriptModule.register_state_dict_pre_hook` \| 可用 \| 在 state_dict 前触发，可接收 module 参数，支持 handle 移除 \| \| `torch.jit.ScriptModule.register_state_dict_post_hook` \| 可用 \| 在 state_dict 后触发，可接收 module 参数，支持 handle 移除 \| 测试日志 ======================================== Python代码多环境测试开始测试脚本: /root/torch-2.11/test_script_module_hooks.py 测试时间: Fri Apr 24 06:37:57 AM UTC 2026 ======================================== 检测到已激活的虚拟环境，先取消激活... ---------------------------------------- 测试环境: torch-2.7.1 虚拟环境: /root/torch-2.7.1/.venv/bin/activate 日志文件: /root/torch-2.11/logs/test_2.7.1_20260424_063757.log ---------------------------------------- 测试环境: torch-2.7.1 退出码: 0 状态: 成功 ✅ 详细日志: /root/torch-2.11/logs/test_2.7.1_20260424_063757.log 完成测试: torch-2.7.1 (退出码: 0) ---------------------------------------- 测试环境: torch-2.8 虚拟环境: /root/torch-2.8/.venv/bin/activate 日志文件: /root/torch-2.11/logs/test_2.8_20260424_063757.log ---------------------------------------- 测试环境: torch-2.8 退出码: 0 状态: 成功 ✅ 详细日志: /root/torch-2.11/logs/test_2.8_20260424_063757.log 完成测试: torch-2.8 (退出码: 0) ---------------------------------------- 测试环境: torch-2.9 虚拟环境: /root/torch-2.9/.venv/bin/activate 日志文件: /root/torch-2.11/logs/test_2.9_20260424_063757.log ---------------------------------------- 测试环境: torch-2.9 退出码: 0 状态: 成功 ✅ 详细日志: /root/torch-2.11/logs/test_2.9_20260424_063757.log 完成测试: torch-2.9 (退出码: 0) ---------------------------------------- 测试环境: torch-2.10 虚拟环境: /root/torch-2.10/.venv/bin/activate 日志文件: /root/torch-2.11/logs/test_2.10_20260424_063757.log ---------------------------------------- 测试环境: torch-2.10 退出码: 0 状态: 成功 ✅ 详细日志: /root/torch-2.11/logs/test_2.10_20260424_063757.log 完成测试: torch-2.10 (退出码: 0) ---------------------------------------- 测试环境: torch-2.11 虚拟环境: /root/torch-2.11/.venv/bin/activate 日志文件: /root/torch-2.11/logs/test_2.11_20260424_063757.log ---------------------------------------- 测试环境: torch-2.11 退出码: 0 状态: 成功 ✅ 详细日志: /root/torch-2.11/logs/test_2.11_20260424_063757.log 完成测试: torch-2.11 (退出码: 0) ======================================== 测试完成总结总测试环境数: 5 完成时间: Fri Apr 24 06:39:05 AM UTC 2026 成功测试数: 5 失败测试数: 0 所有详细日志保存在: /root/torch-2.11/logs 主日志文件: /root/torch-2.11/logs/test_results_20260424_063757.log ======================================== pytorch 2.12 ../root/.local/share/uv/python/cpython-3.12.13-linux-aarch64-gnu/lib/python3.12/multiprocessing/popen_fork.py:66: DeprecationWarning: This process (pid=3985006) is multi-threaded, use of fork() may lead to deadlocks in the child. self.pid = os.fork() [W515 03:02:03.790319610 NPUCachingAllocator.cpp:199] Warning: The current CANN and Soc versions require processing for 32 padding size, with memory allocation. (function operator()) [W515 03:02:11.756102926 IscloseKernelNpu.cpp:34] Warning: Device do not support double dtype of rtol and atol now, dtype cast repalce with float. (function operator()) .................... ---------------------------------------------------------------------- Ran 21 tests in 18.055s OK # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!35723	3 个月前
test_jit_autocast.py	!10958 Add test files: test_flop_counter/test_jit_autocast/test_jit_fuser/... Merge pull request !10958 from yuhaiyan/master-dev3	2 年前
test_jit_fuser.py	!10958 Add test files: test_flop_counter/test_jit_autocast/test_jit_fuser/... Merge pull request !10958 from yuhaiyan/master-dev3	2 年前
test_jit_fuser_te.py	!10958 Add test files: test_flop_counter/test_jit_autocast/test_jit_fuser/... Merge pull request !10958 from yuhaiyan/master-dev3	2 年前
test_matmul_npu.py	!22454 Fix test_matmul_npu Merge pull request !22454 from dilililiwhy/fix_test_matmul_npu	1 年前
test_modules.py	!8912 Synchronize the different test cases for v2.2.0. Merge pull request !8912 from yuhaiyan/master-dev1	2 年前
test_multiprocessing_api.py	skip ci failed testcase Co-authored-by: XianglongZeng<zengxianglong1@huawei.com> # message auto-generated for no-merge-commit merge: !36277 merge v2.12.0 into v2.12.0 skip ci failed testcase Created-by: probiotics_53 Commit-by: XianglongZeng Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 https://gitcode.com/Ascend/op-plugin/pull/4958合入受阻，下掉一批与此次修改无关的门禁失败用例： ## 1. `_afd/test_attention_worker_schedule` `File "/workspace/CODE/pytorch_ut/test/_afd/test_attention_worker_schedule.py", line 4, in <module> import torchair ModuleNotFoundError: No module named 'torchair' [ERROR] ERR99999 UNKNOWN applicaiton exception` ## 2. `_afd/test_ffn_worker_schedule` 同上：`ModuleNotFoundError: No module named 'torchair'` + `ERR99999 UNKNOWN` ## 3. `custom_ops/test_npu_multi_head_attention` `File "/workspace/CODE/pytorch_ut/test/custom_ops/test_npu_multi_head_attention.py", line 98 self.assertRtolEqual(cpu_result, npu_result) AssertionError: result error` 精度比较失败（NPU 和 CPU 计算结果差异超阈值）。 ## 4. `distributed/pipelining/test_schedule`（2 个 error） `ERROR [0.288s]: test_grad_with_split_b_w schedule._load_actions(...) AttributeError: '_PipelineScheduleRuntime' object has no attribute '_load_actions' ERROR [0.058s]: test_grad_with_v_schedule torch.distributed.init_process_group(...)` `_PipelineScheduleRuntime` API 缺失 `_load_actions` 方法，distributed 接口不兼容。 ## 5. `nn/test_module_hooks` `File "/workspace/CODE/pytorch_ut/test/nn/test_module_hooks.py", line 1487 with self.assertRaisesRegex(RuntimeError, "Output 0 of BackwardHookFunctionBackward is " AssertionError: expected: "Output 0 of BackwardHookFunctionBackward is a view..." got: "Output 0 of BackwardHookFunction is a view..."` PyTorch 报错信息中类名变更（`BackwardHookFunctionBackward` → `BackwardHookFunction`），正则匹配失败。 ## 6. `test_multiprocessing_api` `File "/workspace/CODE/pytorch_ut/test/test_multiprocessing_api.py", line 215 self.assertTrue(torch.equal(npu_tensor.cpu(), reconstructed_npu.cpu())) AssertionError: False is not true` 多进程 tensor 序列化/反序列化后数据不一致。 ## 7. `test_ops _complex`（3 个 error，同一根因） `RuntimeError: _index_put_impl_:IndexPutKernelNpuOpApi.cpp:82 call aclnnIndexPutImpl failed, error code is 161002 AclNN_Parameter_Error(EZ1001): Tensor self not implemented for DT_COMPLEX64, should be in dtype support list [DT_FLOAT, DT_INT32, DT_INT64, ...]` ACL 底层 `aclnnIndexPutImpl` 不支持 complex64/complex128 数据类型。 ## 总结 \| # \| 测试 \| 根因 \| 与 0d CPU tensor 修改关系 \| \|---\|------\|------\|:--:\| \| 1 \| attention_worker_schedule \| 环境缺 `torchair` 包 \| 无关 \| \| 2 \| ffn_worker_schedule \| 环境缺 `torchair` 包 \| 无关 \| \| 3 \| multi_head_attention \| 精度差异 \| 无关 \| \| 4 \| test_schedule \| distributed API `_load_actions` 缺失 \| 无关 \| \| 5 \| module_hooks \| PyTorch 错误消息类名变更 \| 无关 \| \| 6 \| multiprocessing_api \| 多进程 tensor 序列化不一致 \| 无关 \| \| 7 \| test_ops _complex \| ACL 不支持 complex dtype \| 无关 \| https://gitcode.com/Ascend/op-plugin/pull/4958 涉及的 15 个 `test_base_ops/` 测试全部 success。 # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】不涉及 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!36277	3 个月前
test_namedtensor.py	!12772 Delete the URLs in code. Merge pull request !12772 from yuhaiyan/master-dev2	2 年前
test_narrow.py	add_test Co-authored-by: sunhaolun<sunhaolun3@huawei.com> # message auto-generated for no-merge-commit merge: !33714 merge master into master add_test Created-by: sunhaolun Commit-by: sunhaolun Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [x] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】为以下API增加用例功能 Tensor.copy_ Tensor.narrow torch.cat torch.chunk torch._chunk_cat # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】新增文件 test_cat.py test_chunk.py test_chunk_cat.py test_copy_.py test_narrow.py # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!33714	4 个月前
test_native_mha.py	!21353 Add test_native_mha.py. Merge pull request !21353 from 刘嘉巍/master	1 年前
test_nestedtensor.py	[feat] add nestedtensor backend Co-authored-by: culechan<chenshuai81@huawei.com> # message auto-generated for no-merge-commit merge: !30982 merge cherry-pick-mr-30981-1771989871865-auto into master [feat] add nestedtensor backend Created-by: culechan Commit-by: culechan Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > (如有)请关联需求文档/issue链接 - [ ] 需求 - [x] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 > 创建Nestedtensor相关的后端，参考quantized后端的实现，因为二者较为类似，因此在文件代码自动生成逻辑部分做了合并重构。 # 【资料变更】 > 不涉及 # 【接口变更】 > 不涉及 # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增已适配UT测试用例看护，并补充测试自验证截图 ![image.png](https://raw.gitcode.com/user-images/assets/7404318/2a752de0-200f-41f4-b2f7-40bb54b8b7e6/image.png 'image.png') # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!30982	6 个月前
test_nn.py	fix(nn): fix test for nn api: torch.nn.ParameterDict, torch.nn.ParameterList, torch.nn.Sequential Co-authored-by: dinglaiping<1016581171@qq.com> # message auto-generated for no-merge-commit merge: !32060 merge fixtest-nn-api-master into master fix(nn): fix test for nn api: torch.nn.ParameterDict, torch.nn.ParameterList, torch.nn.Sequential Created-by: dinglaiping Commit-by: dinglaiping Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">https://gitcode.com/Ascend/pytorch/issues/1585</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] issue/工单 # 【修改方案】 > 请描述修改内容的具体实现，涉及哪些组件之间进行交互，可以用1、2、3、...进行罗列\ > 如果是需求或者重构类的PR，需要补充详细设计文档（说明上下游组件关系、时序图、类图、DFX能力等内容） 1. 2.6.0版本提示skipIfMps找不到，要换成skipIfMPS。 2. 2.7及以上版本skipIfMps找不到，换成skipIfMPS；new_module_tests找不到，需要把导入new_module_tests改成导入get_new_module_tests并在调用的地方换成get_new_module_tests()；tf32_on_and_off找不到，删掉导入，并在调用tf32_is_not_fp32()的地方换成调用torch.cuda.is_tf32_supported() # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图整体结论: 本次验证通过修复官网test_nn.py用例文件，验证3个API，功能正常： ParameterDict：官方用例中键值对索引、新增、删除、遍历等操作均正常，容器内所有参数设备统一为npu:0，无设备漂移问题； ParameterList：官方用例中数字索引、append新增、遍历等操作均正常，参数设备一致性与长度校验均通过，有序管理能力正常； Sequential：官方用例中getitem/setitem/delitem/append/insert/extend/pop等所有核心操作均正常执行，操作后子模块设备均保持为npu:0，动态扩展与元素管理能力完整。运行日志： [root@hostname-fqv42 torchnpuapi]# docker exec -it test bash root@hostname-fqv42:/home# cd /root/torchnpuapi root@hostname-fqv42:~/torchnpuapi# python -m unittest test_nn_npu27.TestNN.test_ParameterList test_nn_npu27.TestNN.test_ParameterDict test_nn_npu27.TestNN.test_Sequential_getitem test_nn_npu27.TestNN.test_Sequential_setitem test_nn_npu27.TestNN.test_Sequential_delitem test_nn_npu27.TestNN.test_Sequential_append test_nn_npu27.TestNN.test_Sequential_insert test_nn_npu27.TestNN.test_Sequential_extend test_nn_npu27.TestNN.test_Sequential_pop -v test_ParameterList (test_nn_npu27.TestNN.test_ParameterList) ... ok test_ParameterDict (test_nn_npu27.TestNN.test_ParameterDict) ... ok test_Sequential_getitem (test_nn_npu27.TestNN.test_Sequential_getitem) ... ok test_Sequential_setitem (test_nn_npu27.TestNN.test_Sequential_setitem) ... ok test_Sequential_delitem (test_nn_npu27.TestNN.test_Sequential_delitem) ... ok test_Sequential_append (test_nn_npu27.TestNN.test_Sequential_append) ... ok test_Sequential_insert (test_nn_npu27.TestNN.test_Sequential_insert) ... ok test_Sequential_extend (test_nn_npu27.TestNN.test_Sequential_extend) ... ok test_Sequential_pop (test_nn_npu27.TestNN.test_Sequential_pop) ... ok ---------------------------------------------------------------------- Ran 9 tests in 0.057s OK root@hostname-fqv42:~/torchnpuapi# # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!32060	5 个月前
test_nnapi.py	!7605 Add UT files Merge pull request !7605 from yuhaiyan/master-dev1	2 年前
test_npu.py	[fix]add getMemoryFraction attribute for torch._C Co-authored-by: weixin_47897441<wuyouqi1@h-partners.com> # message auto-generated for no-merge-commit merge: !30619 merge master into master [fix]add getMemoryFraction attribute for torch._C Created-by: weixin_47897441 Commit-by: weixin_47897441 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > (如有)请关联需求文档/issue链接 - [ ] 需求 - [ ] 问题单 - [x] issue/工单： torch._C不支持_npu_getMemoryFraction - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 1、NPUCachingAllocator中新增getMemoryFraction方法及虚函数声明，Module中注册_npu_getMemoryFraction绑定； 2、在torch_npu.npu和torch_npu.npu.memory中实现并暴露get_per_process_memory_fraction接口，支持指定设备，调用底层 C++ 接口； 3、新增该接口的测试用例； 4、将该接口及底层Cpp绑定加入Dynamo追踪规则，完成适配。 # 【资料变更】 > 不涉及 # 【接口变更】 > 否 # 【功能验证】 > UT用例测试通过：python test_npu.py -v -k test_get_per_process_memory_fraction ![image.png](https://raw.gitcode.com/user-images/assets/7404318/dd7cc124-b3b1-47d1-ab15-ad3798b33115/image.png 'image.png') # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!30619	5 个月前
test_npu_expandable_segments.py	add torch.npu.memory._set_allocator_settings(only support expandable_segments) and testcases Co-authored-by: zhaoyu<nanzhaogang@qq.com> # message auto-generated for no-merge-commit merge: !25488 merge alloctor into master add torch.npu.memory._set_allocator_settings(only support expandable_segments) and testcases 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、增加torch_npu.npu.memory.\_set\_allocator\_settings接口,仅支持expandable_segments:True或expandable_segments:False场景。 2、增加对应的测试用例。 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!25488	10 个月前
test_npu_multinpu.py	!20986 Refactoring device_count Merge pull request !20986 from huangyunlong/2.7ct	1 年前
test_npu_pinned_memory_background_threads.py	host allocator support pinned_use_background_threads Co-authored-by: zhaoyu<nanzhaogang@qq.com> # message auto-generated for no-merge-commit merge: !28786 merge background_thread into master host allocator support pinned_use_background_threads 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: host allocator 支持背景线程 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!28786	7 个月前
test_npu_sanitizer.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!26406	9 个月前
test_ops.py	!9845 update test_trace_rules.py for v2.3 and skip a failed test case. Merge pull request !9845 from yuhaiyan/master-dev4	2 年前
test_ops_fwd_gradients.py	!12772 Delete the URLs in code. Merge pull request !12772 from yuhaiyan/master-dev2	2 年前
test_ops_gradients.py	!12235 UPdate test_ops_gradients and test_ops_fwd_gradients of pt24 Merge pull request !12235 from 张向龙3/ops_grad_24	2 年前
test_ops_jit.py	!10484 Add test files. Merge pull request !10484 from yuhaiyan/master-dev5	2 年前
test_optim.py	!8912 Synchronize the different test cases for v2.2.0. Merge pull request !8912 from yuhaiyan/master-dev1	2 年前
test_out_dtype_op.py	!7765 [UT] add official source testcase for test_out_dtype_op Merge pull request !7765 from 张伟康/pta_master_dev	2 年前
test_reductions.py	!12189 Update test_reductions of pt2.4 Merge pull request !12189 from 张向龙3/reductions_24	2 年前
test_shape_ops.py	!8330 Delete the public URL in code Merge pull request !8330 from yuhaiyan/master-dev6 url fixed url fixed fixed fixed fixed	2 年前
test_sort_and_select.py	!13576 [Fix] Enable UT for isin.Tensor_Tensor. Merge pull request !13576 from 刘嘉巍/master	1 年前
test_sparse.py	!12609 add test_sparse.py Merge pull request !12609 from huangyunlong/2.4sparse	2 年前
test_sparse_coo.py	!14232 [master] fix sparse ut Merge pull request !14232 from DaiFu/2407SparseUT	1 年前
test_sparse_csr.py	Skip failing torch official TestCase tests and disable unsupported test cases Co-authored-by: yuhaiyan<yuhaiyan8@huawei.com> Co-authored-by: yuhaiyan8<yuhaiyan8@h-partners.com> # message auto-generated for no-merge-commit merge: !32140 merge cherry-pick-mr-32139-1774073503832-auto into master Skip failing torch official TestCase tests and disable unsupported test cases Created-by: yuhaiyan Commit-by: yuhaiyan;yuhaiyan8 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 1. 暂时跳过失败用例 2. 已提单跟进修复 # 【资料变更】 > 请确认是否涉及资料变更。如涉及，需要在PR中体现，并简要说明修改内容。如不涉及，需填写“不涉及” # 【接口变更】 > 请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及，需要详细说明接口以及对应的变更内容，同时需要在资料中体现。如不涉及，需填写“不涉及” # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [ ] 代码注释完备，正确记录错误日志 - [ ] 代码实现进行了返回值、空指针等校验 - [ ] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [ ] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!32140	4 个月前
test_tensor_creation_ops.py	!12178 Update test_tensor_creation_ops of pt2.4 Merge pull request !12178 from 张向龙3/creation_ops_24	2 年前
test_testing.py	[feat]TensorTo support preserve_format consistent with GPU Co-authored-by: chenshuai<chenshuai81@huawei.com> # message auto-generated for no-merge-commit merge: !35353 merge cherry-pick-mr-34696-1778548963126-auto into v2.12.0 [feat]TensorTo support preserve_format consistent with GPU Created-by: culechan Commit-by: culechan;chenshuai Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [x] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 1、NPU上tensorto支持preserve_format模式。行为逻辑上对齐pytorch原生框架 2、增加preserve_format模式相关的测试用例 # 【资料变更】不涉及，原生API表格中未体现这部分内容 # 【接口变更】 Tensor.to默认使用且支持preserve_format模式 # 【功能验证】 ![image.png](https://raw.gitcode.com/user-images/assets/7404318/26dfe926-6f3c-4124-9680-486ed681e868/image.png 'image.png') ![image.png](https://raw.gitcode.com/user-images/assets/7404318/a46d3e66-4a4c-4f9f-9531-78b30d665b58/image.png 'image.png') # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!35353	3 个月前
test_torch.py	Update test_torch.py Co-authored-by: yuhaiyan<yuhaiyan8@huawei.com> # message auto-generated for no-merge-commit merge: !27371 merge master-dev2 into master Update test_torch.py Created-by: yuhaiyan Commit-by: yuhaiyan 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!27371	8 个月前
test_torch_npu_init.py	[sync] PR-35820: fix for unsupported_dtype export Co-authored-by: bellatan<tanmei2@huawei.com> # message auto-generated for no-merge-commit merge: !36003 merge sync-pr35820-v2.7.1_torch_npu_init_refactor-to-v2.12.0 into v2.12.0 [sync] PR-35820: fix for unsupported_dtype export Created-by: ascend-ds-bot Commit-by: bellatan Merged-by: ascend-robot Description: ### 1. Origin pull request: https://gitcode.com/Ascend/pytorch/merge_requests/35820 ### 2. Original pull request related issue(s): https://gitcode.com/Ascend/pytorch/issues/1913 ### 3. Original pull request related commit(s): \| Sha \| Datetime \| Message \| \|---\|---\|---\| \|[35b66700](https://gitcode.com/Ascend/pytorch/commit/35b66700591d188d1e6e096a840474e3ea944499)\|2026-05-16 15:00:19 +0800 CST\|fix for unsupported_dtype export<br>\| See merge request: Ascend/pytorch!36003	3 个月前
test_torch_npu_run.py	!13891 add ParallelTcpStore and register to rendezvous Merge pull request !13891 from 叶子凡/master_TCP	2 年前
test_transformers.py	!10598 Delete the URL in code. Merge pull request !10598 from yuhaiyan/master-dev7	2 年前
test_type_promotion.py	!12165 Update test_type_promotion of pt2.4 Merge pull request !12165 from 张向龙3/prom_24	2 年前
test_unary_ufuncs.py	!12150 Update test_unary_ufuncs of pt2.4 Merge pull request !12150 from 张向龙3/unary_ufunc_24	2 年前
test_utils.py	[fix] add _npu_dtype_cast_backward sharding strategy Co-authored-by: zqwenn<zhangqiongwen@huawei.com> # message auto-generated for no-merge-commit merge: merge master_npu_dtype_cast_backsard_register_sharding into master [fix] add _npu_dtype_cast_backward sharding strategy Created-by: zqwenn Commit-by: zqwenn 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!25308	11 个月前
test_view_ops.py	!8289 Add test_view_ops ut Merge pull request !8289 from 王广斌/mv_sort_master	2 年前
testfiles_synchronized.txt	Update onnx_test_common.py Co-authored-by: yuhaiyan<yuhaiyan8@huawei.com> # message auto-generated for no-merge-commit merge: merge master-dev1 into master Update onnx_test_common.py Created-by: yuhaiyan Commit-by: yuhaiyan 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!25485	11 个月前
testfolder_synchronized.txt	!12569 Add test files for the onnx module: test_pytorch_jit_onnx.py/test_pytorch_onnx_no_runtime.py/... Merge pull request !12569 from yuhaiyan/master-dev7	2 年前
torch_npu_schema.json	[feat] Add disable_tensor_unsafe_check option to restart_device Co-authored-by: LiNuohang<linuohang@huawei.com> # message auto-generated for no-merge-commit merge: !36167 merge bz-12 into v2.12.0 [feat] Add disable_tensor_unsafe_check option to restart_device Created-by: LiNuohang Commit-by: LiNuohang Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [x] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】需求背景：推理场景，不需要更新权重数据，因此发生uce故障时，存在只做流恢复，不需要做数据标脏和恢复的场景。当前快恢流恢复和数据标脏通过同一个配置项rebuild_all_resources控制，导致该场景无法使用，需要为数据标脏单独增加配置项。当前现状：当前快恢流恢复和数据标脏通过同一个配置项rebuild_all_resources控制具体设计方案：描述：restart_device接口增加disable_tensor_unsafe_check开关，用于单独控制是否对数据做标脏处理为保证兼容性，默认为False，且仅在rebuild_all_resource为True时生效，即： rebuild_all_resources = True, disable_tensor_unsafe_check = True, 不做数据标脏 rebuild_all_resources = True, disable_tensor_unsafe_check= False, 做数据标脏 rebuild_all_resources = False, disable_tensor_unsafe_check= True, 不做数据标脏 rebuild_all_resources = False, disable_tensor_unsafe_check= False, 不做数据标脏 # 【资料变更】 restart_device接口增加disable_tensor_unsafe_check开关 # 【接口变更】 restart_device接口增加disable_tensor_unsafe_check开关 # 【功能验证】已新增UT # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!36167	3 个月前
url.ini	!12569 Add test files for the onnx module: test_pytorch_jit_onnx.py/test_pytorch_onnx_no_runtime.py/... Merge pull request !12569 from yuhaiyan/master-dev7	2 年前
url.py	!12569 Add test files for the onnx module: test_pytorch_jit_onnx.py/test_pytorch_onnx_no_runtime.py/... Merge pull request !12569 from yuhaiyan/master-dev7	2 年前

简介

test目录为PTA相关的测试用例。

安装依赖

pip3 install -r requirements.txt

补全脚本

该操作需要联网

cd test
bash get_synchronized_files.sh

通过以上操作，会自动补齐testfiles_synchronized.txt和testfolder_synchronized.txt中的文件或文件夹。

跳过失败用例

export DISABLED_TESTS_FILE=./unsupported_test_cases/.pytorch-disabled-tests.json

如果不是在test目录下运行测试用例，需要传入.pytorch-disabled-tests.json的绝对路径。

执行方式

执行单个测试脚本

运行以test开头的文件。以test_autocast.py为例：

方式一：

python test_autocast.py

方式二：

python run_test.py -i test_autocast

说明：部分以test开头的脚本不是直接运行的脚本，比如jit中的测试脚本是通过test_jit.py执行的。

执行具体的用例

通过-k参数传入具体的用例名。以test_autocast.py为例：

方式一：

python test_autocast.py -v -k test_autocast_nn_fp32

方式二：

python run_test.py -v -i test_autocast -- -k test_autocast_nn_fp32

执行全量UT的方式

# 进入到test目录的上一级
cd ../

运行非分布式全量用例：

python ci/access_control_test.py --all

运行分布式全量用例：

python ci/access_control_test.py --distributed

FAQ

报错："dictionary changed size during interation".

如果python 环境是3.8.1版本，报错在unitest/case.py中，可考虑是sys.modules被修改导致的。第三方包可能会有对sys.modules的修改，比如beartype。此问题为python 3.8.1版本/3.9.0版本的已知bug，可按照 https://github.com/python/cpython/issues/73806 中修改方式修改，将 for v in sys.modules.values() 改为 for v in list(sys.modules.values())

可用于复现问题的用例： python test_jit.py -v -k test_annotated_empty_dict
test_public_bindings.py 用例的功能

该用例是为了校验接口的公开规范性，如果该用例报错，请确认报错的接口是否要公开，并按照报错的提示进行修改。