在多核上运行openblas

Question

我正在实现一个简单版本的矩阵每矩阵乘法和矩阵每个向量与openblas与dgemm和dgemv的乘法。我看到openblas只运行在一个核心上。

我试着添加-lpthread进行编译，但没有成功。

我打电话DGEMM和dgemv的方法很简单：

在多个内核上

cblas_dgemv(order, trans, m, n, alpha, a, lda, x, incx, beta, y, incy); 
cblas_dgemm(M, N, K, alpha, A, 1, M, B, 1, K, beta, C, 1, M); 

有没有人成功运行openblas？

Answer 1

您是否尝试过使用environment variables设置线程数？

export OMP_NUM_THREADS=4 

如果这不起作用，你可以设置线程openblas的数​​量通过以下功能使用：

void openblas_set_num_threads(int num_threads); 

比照https://github.com/xianyi/OpenBLAS#set-the-number-of-threads-on-runtime