ubuntu下配置OpenBLAS

OpenBLAS是高度优化的线性代数库。

1.下载编译

git clone https://github.com/xianyi/OpenBLAS.git
cd OpenBLAS
make
sudo make PREFIX=/usr/local/OpenBLAS install

2.CMakeLists

target_link_libraries(untitled1 /usr/local/OpenBLAS/lib/libopenblas.a)
链接blas动态库
target_link_libraries(untitled1 pthread)
线程操作需要pthread

3.例子

void cblas_sgemm(CBLAS_LAYOUT layout, CBLAS_TRANSPOSE TransA,
                 CBLAS_TRANSPOSE TransB, const int M, const int N,
                 const int K, const float alpha, const float *A,
                 const int lda, const float *B, const int ldb,
                 const float beta, float *C, const int ldc);
用于计算C := alpha*op(A)*op(B) + beta*C。
如果alpha=1,beta=0,则计算2个矩阵相乘。
A是 M × K大小,B是K × N大小,C是M × N大小
由于openblas矩阵都是按照一维数组形式存储的,因此参数比较多。
layout:CblasRowMajor  1 2 3                 CblasColMajor   1 4 7
                       4 5 6                                 2 5 8
                       7 8 9                                 3 6 9
CblasRowMajor按行存储,CblasColMajor按列存储
TransA:CblasNoTrans 不进行变换,CblasTrans变为转置,CblasConjTrans变为共轭转置。
TransB:同上
M:A的行数,也就是C的行数,转置后的
N:B的列数,也就是C的列数,转置后的
K:A的列数,也就是B的行数,转置后的
alpha:一般为1
A:矩阵A
B:矩阵B
beta:一般为0
C:矩阵C
lda,ldb,ldc:如果是CblasColMajor则分别表示A,B,C的行,如果是CblasRowMajor,则分别表示A,B,C的列。

4.注意事项

如果应用是多线程的,将会与多线程下的OpenBLAS发生冲突。因此OpenBLAS只能运行在单线程下。可以通过三种方法设置:
1.在环境变量中OPENBLAS_NUM_THREADS=1
2.在应用运行时调用openblas_set_num_threads(1)
3.在编译时选择 make USE_THREAD=0
另外在应用被OpenMP优化时,编译时USE_OPENMP=1
5.测试
10000×10000的矩阵相乘,大概5s

发布了8 篇原创文章 · 获赞 0 · 访问量 6343
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览