已关闭
[Requirement|需求建议]: 新增 aclblasSspr2 算子 arch35 实现 #220
yuyuanfeng创建于  6月25日关闭于  6月26日
yuyuanfeng
yuyuanfeng成员
6月25日 创建

Background(背景信息)

新增 BLAS Level-2 算子 aclblasSspr2 的 arch35 (Ascend 950) 实现。该算子执行对称矩阵 rank-2 更新:A := alpha * x * y^T + alpha * y * x^T + A,数据以 packed 格式存储(列优先)。

Origin(信息来源)

cann 开发者

Benefit / Necessity (价值/作用)

补齐 BLAS Level-2 算子在 arch35 平台上的覆盖,支持 SPR2 运算的 NPU 加速。

Design(设计方案)

采用双路径架构:

  • UB 快速路径:incx==1 && incy==1 且 n 在 UB 容量内时,将 x/y 向量缓存到 UB,多核按列切分处理
  • GM 回退路径:incx!=1 或 incy!=1 或 n 超出 UB 容量时,直接从 GM 读写

包含完整的参数校验、测试套件及 CPU golden 参考实现。

likedislike
yuyuanfengyuyuanfeng成员
6月25日 添加了label:requirement
yuyuanfengyuyuanfeng成员
6月25日 修改了issue 的描述
yuyuanfengyuyuanfeng成员
6月25日 修改了issue 的描述
yuyuanfengyuyuanfeng成员
6月25日 将 yuyuanfeng 设为负责人
CANN-robotCANN-robot成员
6月26日 关闭了 issue
CANN-robotCANN-robot成员
6月26日 添加了label:resolved