Pun*_*han 5 c c++ blas matrix-multiplication
我正在尝试探索 cblas 库;所以我尝试使用 Level3 cblas_sgemm例程进行二维矩阵乘法。
C=(alpha)* A * B + (beta) * C
;; where aplha=1 & beta=0
Run Code Online (Sandbox Code Playgroud)
#define TYPE float
#define A_R 3
#define A_C 2
#define B_R 2
#define B_C 3
int main()
{
TYPE *A=NULL,*B=NULL,*C=NULL;
if(A_C != B_R)
{
printf("\nA(%d,%d) X B(%d,%d) not possible\n",A_R,A_C,B_R,B_C);
exit(1);
}
A=(TYPE*)malloc(A_R*A_C*sizeof(TYPE));
B=(TYPE*)malloc(B_R*B_C*sizeof(TYPE));
C=(TYPE*)malloc(A_R*B_C*sizeof(TYPE));
initM(A,B,C);//initializes matrix
cblas_sgemm(CblasRowMajor, CblasNoTrans,CblasNoTrans,A_R, B_C,A_C, 1.0, A,A_R, B, B_R,0.0, C, A_C);
//multiply(A,B,C);
printf("\n%f\n",C[ (A_R*B_C)-1]);
}
Run Code Online (Sandbox Code Playgroud)
我使用静态库编译了我的程序:
gcc mxm_blas.c -L/CBLAS/lib/ -lcblas -L/BLAS/ -lblas -I/CBLAS/include/ -lgfortran
Run Code Online (Sandbox Code Playgroud)
现在运行代码,我得到:
进入 SGEMM 时,参数号 8 具有非法值
在我看来,麻烦在于声明数组的方式!我见过的大多数示例 将数组声明为:
float A[100][100],b[100][100],C[100][100];
Run Code Online (Sandbox Code Playgroud)
这是错误的原因还是存在程序错误?通过 sgemm API 进行基于 malloc 的矩阵乘法是否不可能?
更新: initM 初始化矩阵:
void initM(TYPE* A,TYPE* B,TYPE* C)
{
long int i,j;
j=A_R*A_C;
for(i=0;i<j;i++) A[i]=2;
j=B_R*B_C;
for(i=0;i<j;i++) B[i]=3;
j=A_R*B_C;
for(i=0;i<j;i++) C[i]=0;
}
Run Code Online (Sandbox Code Playgroud)
错误消息是由sgemm和 产生的cblas_sgemm。的第 8 个参数sgemm是:
子程序SGEMM(TRANSA、TRANSB、M、N、K、ALPHA、A、LDA、B、LDB、BETA、C、LDC)
C 函数 cblas_sgemm 只是调用 Fortran 例程 sgemm 的包装器。基本上它会将一些从 传递到 的参数by value转换by reference。此外,Fortran 例程缺少“RowMajor”参数。
由于您使用的是“RowMajor”存储,因此您应该使用列数作为矩阵的主要维度。
即使我无法测试它,我建议:
cblas_sgemm(CblasRowMajor, CblasNoTrans,CblasNoTrans,A_R, B_C, A_C, 1.0, A, A_C, B, B_C, 0.0, C, B_C);
Run Code Online (Sandbox Code Playgroud)
另外,您可以参考 GSL示例。
| 归档时间: |
|
| 查看次数: |
13555 次 |
| 最近记录: |