已合并
update to support torch-2.15.0 #5719
SCh_zx创建于 3 天前
update to support torch-2.15.0 #5719
已合并
Pull Request已成功合入, 合并人@ascend-robot
(感谢 SCh_zx 的贡献)atomgit-bot
3 天前 评论:
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_NEWEST从V2R14更新为V2R15,标志着 2.15 成为当前支持的最新版本。 - 代码生成脚本版本范围扩展:
gencode.sh中newest_minor_version从 14 更新为 15,使生成脚本能覆盖到 v2r15 的测试文件。 - 配置文件版本列表补齐:
derivatives.yaml和op_plugin_functions.yaml的all_version列表均新增v2.15,确保反向传播规则和算子函数配置涵盖新版本。 - 公共 API 白名单新增 2.15 条目:在
test/allowlist_for_publicAPI.json中为v2.15新增了 11 个允许的公共 API(如npu_gelu_mul、npu_clipped_swiglu、npu_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 版本信息)。


atomgit-bot
3 天前 评论:
3 天前 评论:
ascend-robot
3 天前 评论:
3 天前 评论:
此处折叠了56条消息 查看更多
3 天前 添加了label:approvedlgtm
3 天前 关闭了关联的issue
3 天前 合入了pull request
【合入来源】
【修改方案】
torch更新至2.15
升级以支持编译
【资料变更】
【接口变更】
【功能验证】
【CheckList】