CANN/ops-blas Ssyr算子实现

Ssyr算子实现

【免费下载链接】ops-blas本项目是CANN提供的高性能线性代数计算以及轻量化GEMM调用算子库。项目地址: https://gitcode.com/cann/ops-blas

概述

BLAS Ssyr算子实现。

Ssyr(Symmetric Rank-1 Update)算子实现了单精度向量的外积并将结果加到一个矩阵上，是BLAS基础线性代数库中的核心算子之一。

该算子实现对称秩1更新：A = alpha * x * x^T + A

支持的产品

• Atlas A3 训练系列产品/Atlas A3 推理系列产品
• Atlas A2 训练系列产品/Atlas A2 推理系列产品

目录结构介绍

├── ssyr │ ├── ssyr_kernel.cpp // 算子核函数实现 │ └── ssyr_host.cpp // 算子Host端实现 ├── test/ssyr │ ├── CMakeLists.txt // 编译工程文件 │ ├── README.md // 说明文档 │ └── ssyr_test.cpp // 算子调用样例

算子描述

• 算子功能：
  ssyr算子实现了单精度向量的外积并将结果加到一个矩阵上。对应的数学表达式为：

A = alpha * x * x^T + A

• 对应的接口：

aclblasSsyr(aclblasHandle handle, aclblasFillMode uplo, const int64_t n, const float alpha, uint8_t *x, const int64_t incx, uint8_t *A, const int64_t lda)

• 算子规格：

  算子类型(OpType)	Ssyr
  算子输入	name	shape	data type	format
  	x	[n]	float	ND
  算子输出	A	[n, n]	float	ND
  核函数名	ssyr

• 约束说明：

  • 输入的元素个数n当前覆盖支持[1, 16384]。
  • 算子输入shape为[n]、[n, n]，输出shape为[n, n]。
  • 算子实际计算时，不支持ND高维度运算（不支持维度≥3的运算）。

编译运行

在本样例根目录下执行如下步骤，编译并执行算子。

• 配置环境变量
  请根据当前环境上CANN开发套件包的安装方式，选择对应配置环境变量的命令。

  • 默认路径，root用户安装CANN软件包

    source /usr/local/Ascend/cann/set_env.sh

  • 默认路径，非root用户安装CANN软件包

    source $HOME/Ascend/cann/set_env.sh

  • 指定路径install_path，安装CANN软件包

    source ${install_path}/cann/set_env.sh

• 样例执行

  bash build.sh --ops=ssyr --run # --ops=<算子名> --run可选参数，执行测试样例

  执行结果如下，说明精度对比成功。

  Testing ssyr: Output: ... Golden: ... [Success] Case accuracy is verification passed. [PASS] ssyr_test

【免费下载链接】ops-blas本项目是CANN提供的高性能线性代数计算以及轻量化GEMM调用算子库。项目地址: https://gitcode.com/cann/ops-blas