已关闭
[Requirement|需求建议]: fusedmatmul支持 mm+add/mul融合全模板支持 #3084
xiaotianxiang创建于  6月3日关闭于  6月5日
xiaotianxiang
xiaotianxiang
6月3日 创建

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

Backgroud(背景信息)

FusedMatMul 支持 matmul+add 融合全量模板设计方案
matmul+add融合算子计算公式:Y = A × B + bias + x3(其中x3为elementwise add输入)
matmul+mul融合算子计算公式:Y = (A × B + bias) * x3(其中x3为elementwise add输入)

Origin(信息来源)

非量化matmul

Benefit / Necessity (价值/作用)

期望泛化平均获取到50%以add/mul融合小算子收益。

Design(设计方案)

使用基础API(Basic Level)编程方式实现,适配以下4个模板:
模板名称 FULL_LOAD_MODE L0C2OUT_MODEL 核心类型 说明
ASWT基础模板(AL1全载) MAT_MUL_A_FULL_LOAD(1) MAT_MUL_ON_THE_FLY(0) MIX_AIC_1_2 A矩阵L1全载,on-the-fly输出
ASWT基础模板(BL1全载) MAT_MUL_B_FULL_LOAD(2) MAT_MUL_ON_THE_FLY(0) / MAT_MUL_1V2_ND_ALIG_FIXPIPE(2) MIX_AIC_1_2 B矩阵L1全载,支持fixpipe优化
fixpipe输出优化模板 MAT_MUL_NO_FULL_LOAD(0) / MAT_MUL_B_FULL_LOAD(2) MAT_MUL_1V2_ND_ALIG_FIXPIPE(2) MIX_AIC_1_2 fixpipe bound优化,CV流水覆盖
stream-K模板 MAT_MUL_NO_FULL_LOAD(0) MAT_MUL_ON_THE_FLY(0) / MAT_MUL_1V2_ND_ALIG_FIXPIPE(2) MIX_AIC_1_2 负载均衡stream-K模式

likedislike
oscillatedoscillated成员
6月3日 将 TLE_xtx 设为负责人
CANN-robotCANN-robot成员
6月5日 关闭了 issue
CANN-robotCANN-robot成员
6月5日 添加了label:resolved