已关闭
[Requirement|需求建议]: SubMulConcat内部融合算子适配 #2791
zhongheng创建于  19 天前关闭于  6 天前
zhongheng
19 天前 创建

Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.

一、背景信息 (必填)

SubMulConcat用于融合以下计算:

[
z = concat([x,\ y,\ x-y,\ x*y], axis=-1)
]

本需求为该内部融合算子新增Ascend950实现,补齐FusionPass、Host Tiling、Kernel及构建配置。

二、价值/作用 (必填)

将Sub、Mul和Concat融合为一个Kernel执行,减少算子调度和中间Tensor访存开销,使该融合模式支持Ascend950平台。

三、设计方案 (必填)

3.1 使能方式(涉及哪些框架:如Aclnn直调、Pytorch训练等)
3.2 总体设计
3.2.1 算子支持的数据类型
3.2.2 host侧设计
3.2.3 kernel侧设计
3.3 支持硬件

3.4 算子约束限制

💡 备注(选填)

likedislike
陈思
陈思成员
19 天前 评论:

/assign @zhaozhongyao

likedislike
CANN-robot
CANN-robot成员
19 天前 评论:

Notice

This issue can not be assigned to zhaozhongyao. Please try to assign to the repository members.

likedislike
陈思
陈思成员
18 天前 评论:

/assign @zhaozhongyao

likedislike
CANN-robot
CANN-robot成员
18 天前 评论:

Notice

This issue can not be assigned to zhaozhongyao. Please try to assign to the repository members.

likedislike
陈思
陈思成员
18 天前 评论:

/assign @zhaozhongyao

likedislike
CANN-robot
CANN-robot成员
18 天前 评论:

Notice

This issue can not be assigned to zhaozhongyao. Please try to assign to the repository members.

likedislike
zhongheng
18 天前 评论:

/assign

likedislike
CANN-robotCANN-robot成员
18 天前 将 zhongheng 设为负责人
Zzhongheng
6 天前 issue状态由 进行中 改变为 已完成
Zzhongheng
6 天前 关闭了 issue