Sspr2算子

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

算子概述

Sspr2 算子实现了对称矩阵 packed 格式的秩-2更新操作，核心运算为：A := alpha * x * y^T + alpha * y * x^T + A，其中 A 为 n×n 对称矩阵，以 packed 列优先格式存储，仅上三角或下三角区域被引用和更新。

数学表达式：

A := alpha * x * y^T + alpha * y * x^T + A

包含以下接口：

算子执行接口

aclblasSspr2

产品支持情况

• Ascend 950PR / Ascend 950DT：支持
• Atlas A3 训练系列产品 / Atlas A3 推理系列产品：不支持
• Atlas A2 训练系列产品 / Atlas A2 推理系列产品：不支持

函数原型

aclblasStatus_t aclblasSspr2(aclblasHandle_t handle,
                              aclblasFillMode_t uplo,
                              int n,
                              const float *alpha,
                              const float *x, int incx,
                              const float *y, int incy,
                              float *ap);

参数说明

约束说明

• n >= 0，n == 0 时直接返回成功
• incx != 0 且 incx != INT_MIN
• incy != 0 且 incy != INT_MIN
• 算子输入 shape 为 [n]、[n]，输出 shape 为 [n*(n+1)/2]
• Host 侧不做流同步，调用方需自行管理同步

调用示例

示例代码如下，仅供参考，具体编译和执行过程请参考编译与运行样例。

#include "acl/acl.h"
#include "cann_ops_blas.h"

int main()
{
    aclInit(nullptr);
    aclrtSetDevice(0);

    aclblasHandle_t handle;
    aclblasCreate(&handle);

    int n = 4;
    float alpha = 1.0f;
    int incx = 1, incy = 1;

    size_t size = n * (n + 1) / 2 * sizeof(float);
    float *xDev, *yDev, *apDev;
    aclrtMalloc(&xDev, n * sizeof(float), ACL_MEM_MALLOC_HUGE_FIRST);
    aclrtMalloc(&yDev, n * sizeof(float), ACL_MEM_MALLOC_HUGE_FIRST);
    aclrtMalloc(&apDev, size, ACL_MEM_MALLOC_HUGE_FIRST);

    aclrtMemcpy(xDev, n * sizeof(float), xHost, n * sizeof(float), ACL_MEMCPY_HOST_TO_DEVICE);
    aclrtMemcpy(yDev, n * sizeof(float), yHost, n * sizeof(float), ACL_MEMCPY_HOST_TO_DEVICE);
    aclrtMemsetAsync(apDev, size, 0, size, handle->stream);

    aclblasSspr2(handle, ACLBLAS_UPPER, n, &alpha, xDev, incx, yDev, incy, apDev);

    aclrtSynchronizeStream(handle->stream);

    aclrtFree(xDev);
    aclrtFree(yDev);
    aclrtFree(apDev);

    aclblasDestroy(handle);

    aclrtResetDevice(0);
    aclFinalize();

    return 0;
}

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