使用较少的矩阵与BLAS

时间:2016-10-09 14:54:40

标签: c++ openblas

我对BLAS很新(使用OpenBLAS与C ++和VisualStudio)

我知道dgemm执行 C< - alpha * op(A)* op(B)+ beta * C

我试图保存一些分配: B< - 1 * op(A)* op(B)+ 0 * B

换句话说,将结果放在B矩阵中,

但是使 beta = 0 并在 C 位置重复 B ,结果为零回答

有没有办法让它成功?

我正在使用的代码:

double* A = new double [3*3]; //3 rows x 3 columns

A[0] = 8;
A[1] = 3;
A[2] = 4;
A[3] = 1;
A[4] = 5;
A[5] = 9;
A[6] = 6;
A[7] = 7;
A[8] = 2;

double* v = new double[3]; //3 rows x 1 column

v[0] = 3;
v[1] = 5;
v[2] = 2;

double* foo = new double[3]; //3 rows x 1 column

cblas_dgemm(CblasColMajor, CblasNoTrans, CblasNoTrans,
    3, 1, 3,
    1,
    A, 3,
    v, 3,
    0,
    foo, 3); // makes foo = [41 ; 48 ; 61], **right**

cblas_dgemm(CblasColMajor, CblasTrans, CblasTrans,
    3, 1, 3,
    1,
    A, 3,
    v, 3,
    0,
    v, 3); // makes v = [0 ; 0 ; 0], **wrong**

1 个答案:

答案 0 :(得分:0)

BLAS dgemm function documentation表示只有C矩阵参数用于输入和输出,被操作结果覆盖。由于B仅为输入定义,因此BLAS实现可以假定它不应被修改。

将B和C设置为相同的数据指针可能会触发对您正在使用的实现的一些错误验证,返回归零结果以指示。