已合并
[feat] npu_mhc_pre新增inner_precise支持使能HF32模式 #5661
[feat] npu_mhc_pre新增inner_precise支持使能HF32模式 #5661
已合并
liweijian16创建于 11 天前
liweijian16
11 天前

【合入来源】

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

https://gitcode.com/Ascend/op-plugin/issues/364

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

【修改方案】

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

  1. 新增inner_precise字段;
  2. 新增对aclnnMhcPreV2接口调用,兼容aclnnMhcPre接口

【资料变更】

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

docs/zh/custom_APIs/torch_npu/torch_npu-npu_mhc_pre.md:文档中新增对新可选参数inner_precise的调用说明。

【接口变更】

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

接口新增int inner_precise=0用于指定MhcPre的计算模式,取值0或1,0表示Cube使用FP32模式计算,1表示Cube使用HF32模式计算。

【功能验证】

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

  1. 新精度标准测试结果
    image.png

  2. aclgraph测试结果
    image.png

  3. UT测试结果
    image.png

  4. 异常拦截结果
    image.png

【CheckList】

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

  • 代码注释完备,正确记录错误日志
  • 代码实现进行了返回值、空指针等校验
  • PR标题正确使用类型标签,如:feat、fix、refactor、docs、test等
  • PR持续集成流水线(CI)执行通过,代码检查无异常
likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 liweijian16 的贡献)
Lliweijian16
11 天前 创建了 pull request,commit bf17ee60
ascend-robotascend-robot成员
11 天前 添加了label:stat/needs-squash
ascend-robotascend-robot成员
11 天前 添加了label:ascend-cla/no
ascend-robotascend-robot成员
11 天前 添加了label:needs-issue
此处折叠了118条消息 查看更多
li_jing_hw成员
8 天前 评论:

/approve

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

Pull Request 已合并或已关闭。

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

likedislike