犰狳的eig_sym中的线程安全问题

时间:2017-08-07 14:51:07

标签: c++ thread-safety armadillo eigenvalue eigenvector

我通过eig_sym在犰狳中进行特征分解存在问题。当我尝试并行计算多组特征值和特征向量时,特征向量不时

  • 不正交
  • 未正常化
  • 甚至不是所讨论矩阵的特征向量。

如果每次只运行一次计算,这个问题就会消失(所以这似乎是一些线程安全问题)。很快,当两个计算并行运行时,问题再次出现。奇怪的是,在每种情况下,特征值似乎都是正确的。

//compile with: g++ -std=c++11 -pthread -larmadillo -o evecs armadillo_evecs.cpp
#include <iostream>
#include <armadillo>
#include <assert.h>
#include <future>
#include <vector>
using namespace std;

void testcalc() {
    // set up random symmetric matrix
    arma::mat r = arma::randu<arma::mat> (100, 100);
    r  = r.t() * r;

    arma::vec eval;
    arma::mat evec;
    // calculate eigenvalues and -vectors
    assert(arma::eig_sym(eval, evec, r));
    arma::mat test = evec.t() * evec;

    // Check whether eigenvectors are orthogonal, (i. e. matrix 'test' is diagonal)
    assert(arma::norm(test - arma::diagmat(test)) < 1.0e-10);
}


int main() {
    // start 100 eigenvalue (+vector) calculations
    vector<future<void>> fus;
    for (size_t i = 0; i < 100; i++) {
        // try parallel evaluation ... fails sometimes 
        fus.push_back(async(launch::async, &testcalc));

        // try sequential evaluation ... works fine
        // future<void> f = async(launch::async, &testcalc);
        // f.get(); // Wait until calculation has finished, before starting new one
    }

    // wait until calculations have finished
    for(auto it = fus.begin(); it != fus.end(); it++) {
        it->get();
    }

    return 0;
}

所以在断言上面的代码中

assert(arma::norm(test - arma::diagmat(test)) < 1.0e-10);

有时会失败。这可能是底层库的问题(我读过lapack有一些线程安全问题)?我真的不知道,从哪里开始寻找。

1 个答案:

答案 0 :(得分:1)

使用已经由底层库提供的并行化更容易,更安全,而不是“滚动自己的并行化”。

因此,不要使用引用BLAS和LAPACK,而是使用多线程版本,如OpenBLASIntel MKL。有关详细信息,请参阅犰狳FAQ