已合并
fix: 完善 HostCPU 自定义算子的引擎选择 #4630
duhua创建于 2 天前
fix: 完善 HostCPU 自定义算子的引擎选择 #4630
已合并
duhua创建于 2 天前
duhua
duhua成员
2 天前

Pull Request

描述

完善 HostCPU 自定义算子的引擎选择,保证 HostCPU 自定义实现不会被普通 HostCPU 引擎覆盖:

  • EnginePlacer 节点指定engine时,不会进入DNNEngineManager 选择引擎,因此补充逻辑,在完成引擎选择后,仅对已选择 DNN_VM_HOST_CPU 且存在 HostCPU 自定义实现的节点切换为 DNN_VM_CUSTOM
  • 保留 DNNEngineManager 原有的显式 Host 放置、直接引擎选择及 HostCPU custom 回退逻辑,兼容无普通 HostCPU kernel store 的场景。
  • 增加引擎字段、引擎属性和设备引擎保护场景的 UT。
  • 修正融合示例中 GEMM scalar 常量的类型为 float,适配当前算子优化支持范围。

变更类型

关联的Issue

如何测试

  • 已补充 EnginePlacer HostCPU custom 相关 UT。

核对清单

其他信息

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 duhua 的贡献)
duhuaduhua成员
2 天前 添加了label:enhancement
duhuaduhua成员
2 天前 创建了 pull request,commit 10c4708c
CANN-robotCANN-robot成员
2 天前 添加了label:cann-cla/yes
CANN-robot
CANN-robot成员
2 天前 评论:

Thanks for your pull-request.
The full list of commands accepted by me can be found at here.
You can get sig-info at here.
You can self-configure the PR merge rules for this repository. For more details, please refer to Here.
For more, you also can visit HICANN.


PR Approval Progress

Congratulations! All modules have met the lgtm and approve requirements.

Module Approval Details

module lgtm status approve status
repo-cann/ge 张德鹏, stevenaw0, kobemini, yangyongqiang0606 (4/2) 张德鹏 (1/1)

💡 Tip:

  • Committer can comment /approve or /lgtm
  • Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

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

likedislike
CANN-robotCANN-robot成员
2 天前 将yangyongqiang0606,kobemini,ClarkXie,tangqunzhang,xchu42,hugo111,lining23666,fu-jun2,depeng1994,wqtshg_wt,peiyang,shengnan666,zhangfan_hanq,zhanj,zhujingjing,Mexyy,wangxiaotian995,stevenaw0,yuht9设为评审人
此处折叠了79条消息 查看更多
CANN-robotCANN-robot成员
2 天前 添加了label:lgtm
黄桂军
黄桂军成员
2 天前 评论:

/lgtm

likedislike
张德鹏成员
2 天前 评论:

/approve

likedislike
CANN-robotCANN-robot成员
2 天前 添加了label:approved
CANN-robotCANN-robot成员
2 天前 合入了pull request