已关闭
[Requirement|需求建议]: BlockSparseAttention新增接口aclnnBlockSparseAttentionV2 #2320
tangkaidi创建于  5月9日关闭于  6月1日
tangkaidi成员
5月9日 创建

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

Backgroud(背景信息)

BlockSparseAttention新增接口aclnnBlockSparseAttentionV2。
新增qDequantScale、kDequantScale、vDequantScale参数,支持float8量化功能。

Benefit / Necessity (价值/作用)

降低内存占用和访问,相较非量化场景达到性能提升。

Design(设计方案)

量化级别为Per-Block。Block大小与BlockShape大小保持一致。
MM1阶段结束时随路反量化。
VEC1阶段末尾时对P矩阵量化。
MM2阶段结束时随路反量化。
VEC2阶段末尾时对O矩阵反量化。

likedislike
Ttangkaidi成员
5月9日 添加了label:requirement
Ttangkaidi成员
5月9日 将 tangkaidi 设为负责人
Ttangkaidi成员
5月9日 关联了pull request:新增aclnnBlockSparseAttentionV2接口,相关代码与资料
leiqingji成员
5月11日 评论:

/assign @tangkaidi

likedislike
CANN-robot
CANN-robot成员
5月11日 评论:

Notice

This issue is already assigned to tangkaidi. Please do not assign repeatedly.

likedislike
Lleiqingji成员
5月11日 关联了看板:Transformer
Ttangkaidi成员
5月30日 删除了关联的pull request:aclnnBlockSparseAttentionV2资料
Ttangkaidi成员
5月30日 关联了pull request:新增aclnn接口aclnnBlockSparseAttentionV2,适配fp8量化的BlockSparseAttention算子
Ttangkaidi成员
5月30日 修改了issue 的描述
Ttangkaidi成员
5月30日 修改了issue 的描述
Ttangkaidi成员
5月30日 删除了关联的pull request:新增aclnn接口aclnnBlockSparseAttentionV2,适配fp8量化的BlockSparseAttention算子
Ttangkaidi成员
5月30日 关联了pull request:新增aclnn接口aclnnBlockSparseAttentionV2,适配fp8量化的BlockSparseAttention算子
CANN-robotCANN-robot成员
6月1日 关闭了 issue
CANN-robotCANN-robot成员
6月1日 添加了label:resolved