已合并
compat(v2.14): register v2.14, follow upstream #187357 #5644
SCh_zx创建于 22 天前
compat(v2.14): register v2.14, follow upstream #187357 #5644
已合并
SCh_zx创建于 22 天前
SCh_zx成员
22 天前

【合入来源】

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

【修改方案】

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

torch_npu CI 流水线编译阶段失败,错误位于自动生成的 torch_npu/csrc/aten/RegisterNPU.cpp 中 _scaled_grouped_mm_v2 算子的 wrapper 函数。codegen 生成的 wrapper 将 scale_a 和 scale_b 参数类型生成为 const at::ITensorListRef &,但 op-plugin 中 ScaledGroupMmV2KernelNpuOpApi.cpp 的实现仍使用 c10::ArrayRefat::Tensor(即 at::TensorList),两者不兼容导致编译失败。

【资料变更】

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

【接口变更】

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

【功能验证】

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

【CheckList】

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

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 SCh_zx 的贡献)
SSCh_zx成员
22 天前 创建了 pull request,commit cf70161c
SSCh_zx成员
22 天前 关联了issue:[Bug] torch_npu编译失败:_scaled_grouped_mm_v2 structured化导致ITensorListRef类型不匹配
atomgit-bot
atomgit-bot
22 天前 评论:

变更摘要

此 PR 主要跟进上游 #187357 的接口变更,对 ScaledGroupMmV2KernelNpuOpApi.cpp 中的 _scaled_grouped_mm_v2 函数签名进行兼容性适配:将 scale_ascale_b 两个参数类型从 c10::ArrayRef<at::Tensor> 替换为 const at::ITensorListRef&,以匹配上游 PyTorch 对张量列表引用的新接口规范。

主要改动

  • scale_a 参数类型变更:将函数参数 scale_a 的类型从 c10::ArrayRef<at::Tensor> 改为 const at::ITensorListRef&,以适配上游对张量列表引用的统一封装。
  • scale_b 参数类型变更:将函数参数 scale_b 的类型从 c10::ArrayRef<at::Tensor> 改为 const at::ITensorListRef&,与 scale_a 保持一致,确保接口兼容性。
likedislike
atomgit-bot
atomgit-bot
22 天前 评论:

代码审查

我已审查完毕。以下是最终结论。


审查总结

优先级 数量
P0 1
P1 0
P2 0
P3 0

已审查文件:

  • op_plugin/ops/opapi/ScaledGroupMmV2KernelNpuOpApi.cpp发现 1 个 P0 问题

整体风险评估:此变更不可合入。 DIFF 仅将 _scaled_grouped_mm_v2 函数签名中的 scale_a / scale_b 参数类型从 c10::ArrayRef<at::Tensor> 改为 const at::ITensorListRef&,但未在函数体中添加 materialize 代码。函数体第 174、177 行将 scale_a / scale_b 传递给期望 c10::ArrayRef<at::Tensor>& 的 matcher(check_rowwise_recipe / check_mxfp8_recipe),由于 ITensorListRef 无法隐式转换为 ArrayRef,必然导致编译失败。修复需参照同仓库 ScaledMmV2KernelNpuOpApi.cpp_scaled_mm_v2 的完整 materialize 模式(第 333–338 行)。

类型 数量
🔴 阻塞 1
🟡 建议 0

⛔ 需要修改

likedislike
此处折叠了142条消息 查看更多
梁松伟
梁松伟成员
21 天前 评论:

/approve

likedislike
ascend-robotascend-robot成员
21 天前 添加了label:approvedlgtm
ascend-robotascend-robot成员
21 天前 关闭了关联的issue
ascend-robotascend-robot成员
21 天前 合入了pull request
ascend-robot
ascend-robot成员
21 天前 评论:

Pull Request 已合并或已关闭。

If you want to solve this problem, you can click here to do it in the FAQs.

likedislike