【合入来源】

关联 issue

• 任务 Issue：https://gitcode.com/Ascend/pytorch/issues/2677
• 说明性 Issue：https://gitcode.com/Ascend/pytorch/issues/4679

【修改方案】

修改说明

本 PR 在 v2.11.0 分支新增 test/test_tensor_setstate_api_completion.py，直接验证 torch.Tensor.__setstate__ 的核心行为：

• 叶子 Tensor 调用 __setstate__ 后可以恢复 requires_grad 状态；
• 非叶子 Tensor 调用 __setstate__ 时抛出与原生 PyTorch 一致的异常；
• 测试通过 torch.accelerator.current_accelerator() 选择当前设备，NPU 环境实际使用 NPU。

修改文件：

test/test_tensor_setstate_api_completion.py

涉及 API

torch.Tensor.__setstate__

社区用例覆盖情况

PyTorch 社区用例主要通过序列化、JIT 或 pickle 等间接路径触发 __setstate__，没有面向该接口的直接行为测试。因此本 PR 增加直接 API 调用覆盖，验证 Ascend NPU 上的行为与原生 PyTorch 保持一致。

NPU 适配

测试不使用 transfer_to_npu，也不包含 tensor.cuda()、torch.cuda.* 或 device="cuda" 等 CUDA 迁移路径。

device_type = acc.type if (acc := torch.accelerator.current_accelerator()) else "cpu"

无加速器环境回退到 CPU 仅用于保证测试文件可导入；NPU 环境通过 accelerator=npu 和 NPU_AVAILABLE=True 确认实际走 NPU 路径。

测试核心逻辑

tensor = torch.ones(2, device=device_type)
self.assertFalse(tensor.requires_grad)

tensor.__setstate__((True, None, None))
self.assertTrue(tensor.requires_grad)

with self.assertRaisesRegex(
    RuntimeError, "__setstate__ can be only called on leaf Tensors"
):
    (tensor + 1).__setstate__((False, None, None))

【资料变更】

已通过独立 doc PR 在 docs/zh/api/native_api/private_apis.md 中补充 API 记录。

【接口变更】

不涉及 API 签名、实现或行为变更，本 PR 仅补充测试覆盖。

【功能验证】

910B3 测试结果

COMMAND: /workspace/venvs/torch2110/bin/python /workspace/tensor_setstate_910B3_tests_20260908/test_tensor_setstate_api_completion_v2110.py -v
PWD: /tmp
DATE: 2026-09-08 14:12:11 CST
test_tensor_setstate (__main__.TestTensorSetstateApiCompletion.test_tensor_setstate) ... ok

----------------------------------------------------------------------
Ran 1 test in 1.422s

OK
EXIT_CODE: 0

Ascend950PR 测试结果

COMMAND: /tmp/benchmark-api-venvs/torch211/bin/python /tmp/tensor_setstate_950pr_tests_20260908/test_tensor_setstate_api_completion.py -v
PWD: /tmp
DATE: 2026-09-08 14:22:23 UTC
test_tensor_setstate (__main__.TestTensorSetstateApiCompletion.test_tensor_setstate) ... [W908 14:22:27.764007073 NPUCachingAllocator.cpp:201] Warning: The current CANN and Soc versions require processing for 32 padding size, with memory allocation. (function operator()) ok

----------------------------------------------------------------------
Ran 1 test in 1.964s

OK
EXIT_CODE: 0

950PR 硬件为 Ascend950PR，npu-smi 25.7.rc1，Health 为 OK。

【CheckList】

• 代码注释完备，正确记录错误日志
• 代码实现进行了返回值、空指针等校验
• PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等
• PR持续集成流水线（CI）执行通过，代码检查无异常（PR 创建并通过 CI 后勾选）