已合并
update to support torch-2.15.0 #5719
SCh_zx创建于 3 天前
update to support torch-2.15.0 #5719
已合并
SCh_zx创建于 3 天前
SCh_zx成员
3 天前

【合入来源】

如有社区issue,请关联issue链接
请勿携带内部流程信息(需求链接、问题单、内部issue等)

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

【修改方案】

请描述修改内容的具体实现,涉及哪些组件之间进行交互,可以用1、2、3、...进行罗列
如果是需求或者重构类的PR,需要补充详细设计文档(说明上下游组件关系、时序图、类图、DFX能力等内容)

torch更新至2.15
升级以支持编译

【资料变更】

请确认是否涉及资料变更。如涉及,需要在PR中体现,并简要说明修改内容。如不涉及,需填写“不涉及”

【接口变更】

请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及,需要详细说明接口以及对应的变更内容,同时需要在资料中体现。如不涉及,需填写“不涉及”

【功能验证】

说明测试场景,测试方法。如果本次测试方式与常规单元测试不同,请详细说明您的测试步骤
新增/变更内容是否已新增/适配UT测试用例看护,并补充测试自验证截图

【CheckList】

PR提交人对以下CheckList自检项进行全量自检,自检通过或不涉及,均修改 [ ] 为 [x]

  • 代码注释完备,正确记录错误日志
  • 代码实现进行了返回值、空指针等校验
  • PR标题正确使用类型标签,如:feat、fix、refactor、docs、test等
  • PR持续集成流水线(CI)执行通过,代码检查无异常
likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 SCh_zx 的贡献)
SSCh_zx成员
3 天前 创建了 pull request,commit 511380d4
SSCh_zx成员
3 天前 关联了issue:[Installation]: update to support newest torch version
atomgit-bot
atomgit-bot
3 天前 评论:

变更摘要

此 PR 主要用于将项目升级以支持 PyTorch 2.15.0 版本。核心改动包括:在版本配置文件中新增 v2.15 版本条目,更新 Version.h 中的最新版本宏定义,为 2.15 版本生成测试所需的不支持算子信息文件,以及在公共 API 白名单中添加 2.15 版本允许的 API 列表。同时清理了 torchnpugen/context.py 中过时的注释。

主要改动

  • 版本宏与最新版本号更新:在 op_plugin/utils/Version.h 中新增 V2R15 宏(值为 14),并将 VERSION_NEWESTV2R14 更新为 V2R15,标志着 2.15 成为当前支持的最新版本。
  • 代码生成脚本版本范围扩展gencode.shnewest_minor_version 从 14 更新为 15,使生成脚本能覆盖到 v2r15 的测试文件。
  • 配置文件版本列表补齐derivatives.yamlop_plugin_functions.yamlall_version 列表均新增 v2.15,确保反向传播规则和算子函数配置涵盖新版本。
  • 公共 API 白名单新增 2.15 条目:在 test/allowlist_for_publicAPI.json 中为 v2.15 新增了 11 个允许的公共 API(如 npu_gelu_mulnpu_clipped_swiglunpu_fused_floyd_attention 等),与前一版本白名单保持一致。
  • 2.15 不支持的算子信息文件初始化:新增 test/test_v2r15_ops/unsupported_ops_info.yaml,仅包含 base 注释占位,待后续补充 2.15 版本不支持的具体算子信息。
  • 过时注释清理:移除 torchnpugen/context.py 中关于 _scaled_mm_v2 的已过时注释(原注释引用了上游 pytorch#185272 和 2.14 版本信息)。
likedislike
atomgit-bot
atomgit-bot
3 天前 评论:

代码审查

✅ 未发现问题

likedislike
此处折叠了56条消息 查看更多
ascend-robotascend-robot成员
3 天前 添加了label:approvedlgtm
ascend-robotascend-robot成员
3 天前 关闭了关联的issue
ascend-robotascend-robot成员
3 天前 合入了pull request
SSCh_zx成员
3 天前 修改了pull request 的描述
SSCh_zx成员
3 天前 修改了pull request 的描述