已合并
test: add MLIR non-fallback kernel regression test #38520
yvjc创建于 6月15日
test: add MLIR non-fallback kernel regression test #38520
已合并
yvjc创建于 6月15日
已删除 :fusionTest合入到Ascend/pytorchv2.7.1
yvjc成员
6月15日

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

【合入来源】

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

【修改方案】

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

解决MLIR多进程编译问题。将导致出现问题的语句改为延迟加载形式。
新增 MLIR 后端 kernel 编译回归 UT:

  1. 构造基础 pointwise 融合计算,并通过 torch.compile 使用 MLIR 后端编译执行。
  2. Hook NpuMlirCompiler.run,记录实际执行 kernel 的 is_fallback_kernel 状态。
  3. 断言执行过程中不存在 fallback kernel,防止 MLIR kernel 编译失败后静默回退。
  4. 将编译后的 NPU 结果与 CPU eager 结果进行精度对比。

【资料变更】

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

不涉及

【接口变更】

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

不涉及

【功能验证】

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

ut验证通过

【CheckList】

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

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 yvjc 的贡献)
Yyvjc成员
6月15日 创建了 pull request,commit a2dd5c03
ascend-robotascend-robot成员
6月15日 添加了label:ascend-cla/yes
ascend-robotascend-robot成员
6月15日 添加了label:needs-issue
ascend-robot
ascend-robot成员
6月15日 评论:

CLA Signature Pass

yvjc, thanks for your pull request. All authors of the commits have signed the CLA. 👍

likedislike
此处折叠了181条消息 查看更多
ascend-robotascend-robot成员
6月18日 添加了label:approved
daifu1234567成员
6月18日 评论:

/lgtm

likedislike
ascend-robotascend-robot成员
6月18日 添加了label:lgtm
ascend-robotascend-robot成员
6月18日 合入了pull request
ascend-robot
ascend-robot成员
6月18日 评论:
流水线 pytorch_gitcode_PR_multiVersion#10944 [ commitID:c5ae3176 ] 运行失败
likedislike