CANN/ops-blas strttp算子

strttp 算子

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

功能描述

三角常规格式转压缩格式：将常规二维三角矩阵压缩为 packed format 存储。

• uplo == ACLBLAS_LOWER：将A的下三角复制到AP
• uplo == ACLBLAS_UPPER：将A的上三角复制到AP

目录结构

blas/trttp/ ├── README.md # 说明文档 └── arch35/ # 算子实现（trttp=操作族, strttp=float32, arch35=ascend950） ├── strttp_host.cpp ├── strttp_kernel.cpp └── strttp_tiling_data.h test/trttp/ # 测试代码 ├── CMakeLists.txt ├── strttp_param.h ├── strttp_golden.h └── arch35/ ├── strttp_npu_wrapper.h ├── strttp_test.cpp └── strttp_test.csv

接口定义

aclblasStatus_t aclblasStrttp(aclblasHandle_t handle, aclblasFillMode_t uplo, int n, const float *A, int lda, float *AP);

参数说明

编译运行

bash build.sh --ops=strttp --run --soc=ascend950

测试说明

ST 采用 GTest 参数化 +strttp_test.csv，BlasTest<StrttpParam>fixture，精度模式为EXACT。

注意：makeBlasArray的 size 参数为int64_t，调用时需显式转换：makeBlasArray(static_cast<int64_t>(p.lda) * p.n, p.a)，确保负值 n 正确返回空数组。

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