变更摘要

此 PR 恢复了对 torch_npu.utils.load 中 legacy（非 zip）加载分支的修复：将 _legacy_load 的第二个参数从硬编码的 "cpu" 改回透传 map_location，使加载结果能按 map_location 正确映射到目标设备，修复 torch.compile 缓存加载在 NPU 上的功能问题。同时恢复了对应的单测用例 test_legacy_load_maplocation。

主要改动

• torch_npu/utils/serialization.py 修复 map_location 透传：在 load 函数的 legacy 加载分支中，将 _legacy_load(opened_file, "cpu", ...) 改为 _legacy_load(opened_file, map_location, ...)，使 map_location 参数在非 (torch.device, str) 类型分支下被正确传递，而非被硬编码为 "cpu"。
• 新增 test_legacy_load_maplocation 测试用例：在 test/npu/test_serialization.py 中新增参数化测试，使用 @parametrize("map_kind", ["none", "callable", "dict"]) 覆盖 None、可调用对象、字典三种 map_location 形式，验证 legacy 加载模式下结果能正确映射到 npu:0 设备。
• 启用参数化测试支持：新增 parametrize 和 instantiate_parametrized_tests 的导入，并在模块级调用 instantiate_parametrized_tests(TestSerialization) 以激活参数化测试。

openLiBing可接管PR, PR详情链接:
https://www.openlibing.com/apps/transitionPrDetail?owner=Ascend&repo=pytorch&number=43272&platform=gitcode&codeHostingPlatformFlag=gitcode

代码审查

审查总结

审查覆盖情况

发现统计

• P0: 0
• P1: 0
• P2: 0
• P3: 1

整体风险评估

低风险。 核心修复（第 339 行将 "cpu" 改为 map_location）正确解决了 map_location=None（以及 Callable/Dict）在 legacy 加载分支被错误强制为 CPU 的 bug，与 issue #3005 描述的问题根因一致。新增的单测 test_legacy_load_maplocation 通过 @parametrize 覆盖了 None / Callable / Dict 三种 map_location 类型，结构合理。唯一可改进项是第 315–320 行的遗留警告消息与修复后的行为不一致，属于 P3 级别的文档维护问题，不影响功能正确性。

⚠️ 已识别出整体风险，但无法提取行内评论，请参考整体评估。

/approve

/lgtm