0

我对 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**
4

1 回答 1

0

BLAS dgemm 函数文档指出，只有 C 矩阵参数同时用于输入和输出，被运算结果覆盖。由于 B 仅用于输入，BLAS 实现可以假定它不应该被修改。

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

于 2016-10-09T22:09:24.727 回答