【合入来源】

https://gitcode.com/Ascend/pytorch/issues/2924

• 需求
• 问题单
• issue/工单
• 重构优化
• 资料更新

【修改方案】

一、API 功能

torch._utils._get_available_device_type 查询当前可用加速设备类型；Torch-NPU 环境预期返回 "npu"。

二、PyTorch 社区用例情况与新增必要性

已在 PyTorch 官方社区和 Torch-NPU 原有测试中检索 torch._utils._get_available_device_type，没有发现针对该 API 返回值的直接断言。该 API 是私有工具函数，不涉及 Tensor 或算子计算。

因此本 PR 在 torch-npu test 目录新增独立、可扩展的测试文件：

test/test_utils_api.py::TestTorchUtilsAPIs::test_get_available_device_type

用例只直接断言 torch._utils._get_available_device_type() 返回 "npu"，不增加冗余环境探测、Tensor 创建或异常包装。

三、NPU 适配

3.1 API 适配

Torch-NPU 已将 PrivateUse1 后端注册为 npu，API 实际返回 "npu"，无需修改 torch_npu/ 源码。

3.2 测试用例适配

该 API 是非计算类工具函数，不涉及 Tensor，因此新增测试不需要创建 NPU Tensor；测试通过返回值直接验证 NPU 后端注册结果。

【资料变更】

无资料文件变更。该 API 是 PyTorch 私有接口，按照任务规范无需补充资料。

【接口变更】

无客户可见接口变更，仅新增直接测试。

【功能验证】

验证环境：PyTorch 2.12.0、torch-npu 对应版本、Ascend910B4。

python test/test_utils_api.py -v

test_get_available_device_type (__main__.TestTorchUtilsAPIs.test_get_available_device_type) ... ok

----------------------------------------------------------------------
Ran 1 test in 0.205s

OK

返回值为 "npu"，表明测试实际加载了 NPU PrivateUse1 后端；没有跳过新增测试。

【CheckList】

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