学习BLAS库 -- TRMV

最新推荐文章于 2022-02-06 16:51:32 发布

原创

最新推荐文章于 2022-02-06 16:51:32 发布 · 972 阅读

1 ·

CC 4.0 BY-SA版权

本文详细介绍了BLAS库中的TRMV函数，包括其语法、功能，即进行三角矩阵向量乘法。同时，文章通过OpenBLAS的trmv方法测试了性能基准，展示了实际应用中的效果。

函数语法:

XTRMV( UPLO, TRANS, DIAG, A, LDA, X, INCX)

功能：

Triangular matrix vector multiply

Open BLAS trmv methods testing benchmark

//
// TRMV   x <- Ax
//
/*
 * void cblas_strmv(
 OPENBLAS_CONST enum CBLAS_ORDER order,
 OPENBLAS_CONST enum CBLAS_UPLO Uplo,
 OPENBLAS_CONST enum CBLAS_TRANSPOSE TransA,
 OPENBLAS_CONST enum CBLAS_DIAG Diag,
 OPENBLAS_CONST blasint N,
 OPENBLAS_CONST float *A,
 OPENBLAS_CONST blasint lda,
 float *X,
 OPENBLAS_CONST blasint incX);
 */
void strmv_test() {
	int n = 2;

	float *x = (float*) calloc(n, sizeof(float));
	float *b = (float*) calloc(n, sizeof(float));
	float *A = (float*) calloc(n * n, sizeof(float));
	float *p = (float*) calloc(n, sizeof(float));
	float *r = (float*) calloc(n, sizeof(float));

	A[0] = 4;
	// A[1] = 0;
	A[2] = 1;
	A[3] = 3;

	b[0] = 0;
	b[1] = 0;

	x[0] = 2;
	x[1] = 1