Closed
[Requirement|需求建议]: SFA supports arch92 #6823
wangyinchu1Add on  16 days agoClosed at  16 days ago
wangyinchu1
wangyinchu1Collaborators
16 days ago Create

Backgroud(背景信息)

SFA supports arch92

Benefit / Necessity (价值/作用)

Design(设计方案)

likedislike
wangyinchu1wangyinchu1Collaborators
16 days ago Added label: requirement
游震Collaborators
16 days ago Comments:

/assign @wangyinchu1

likedislike
CANN-robotCANN-robotCollaborators
16 days ago Assign wangyinchu1 as responsible person
周云飞
周云飞Collaborators
16 days ago Comments:

感谢提出 SFA 支持 arch92 的需求。

目前可确认,仓库中已有 KV 量化稀疏注意力(KvQuantSparseFlashAttention)的 arch92 内核代码,但这不足以说明您所指的 SFA 接口已经支持 arch92,也不能据此确定可用版本及输入限制。

请补充 SFA 对应的算子全名或调用接口,以及计划使用的 CANN 版本,以便明确适配对象、判断现有实现是否覆盖您的场景。如果有具体使用需求,也欢迎提供输入数据类型和典型形状,便于进一步评估支持范围。

一个可行的方向是先确认目标接口在 arch92 上的编译、分发和计算路径是否完整,再评估需要补充的适配及验证范围。

likedislike
CANN-robotCANN-robotCollaborators
16 days ago Close issue
CANN-robotCANN-robotCollaborators
16 days ago Added label: resolved
周云飞
周云飞Collaborators
2 days ago Comments:

已合入 PR #13658,为 SFA 增加 arch92 相关支持:新增 arch92 内核公共定义和 Ascend 960DT 算子配置;在接口与 Tiling 中适配 DAV_9201、DAV_9202,包括 sinks 输入、参数校验、工作空间及核数计算等。

关联 PR:PR #13658

likedislike