在我的代码中，我有一个循环，我在其中构建和确定线性系统并尝试解决它:

#pragma omp parallel for
for (int i = 0; i < n[0]+1; i++) {
    for (int j = 0; j < n[1]+1; j++) {
        for (int k = 0; k < n[2]+1; k++) {
            arma::mat A(max_points, 2);
            arma::mat y(max_points, 1);
            // initialize A and y

            arma::vec solution = solve(A,y);
        }
    }
}

有时，程序非常随机地挂起，或者解 vector 中的结果为 NaN。如果我这样做:

arma::vec solution;
#pragma omp critical 
{
    solution = solve(weights*A,weights*y);
}

那么这些问题似乎不再发生了。

当它挂起时，它这样做是因为一些线程在 OpenMP 屏障处等待:

Thread 2 (Thread 0x7fe4325a5700 (LWP 39839)):
#0  0x00007fe44d3c2084 in gomp_team_barrier_wait_end () from /usr/lib64/gcc-4.9.2/lib64/gcc/x86_64-redhat-linux-gnu/4.9.2/libgomp.so.1
#1  0x00007fe44d3bf8c2 in gomp_thread_start () at ../.././libgomp/team.c:118
#2  0x0000003f64607851 in start_thread () from /lib64/libpthread.so.0
#3  0x0000003f642e890d in clone () from /lib64/libc.so.6

其他线程被卡在 Armadillo 内部:

Thread 1 (Thread 0x7fe44afe2e60 (LWP 39800)):
#0  0x0000003ee541f748 in dscal_ () from /usr/lib64/libblas.so.3
#1  0x00007fe44c0d3666 in dlarfp_ () from /usr/lib64/atlas/liblapack.so.3
#2  0x00007fe44c058736 in dgelq2_ () from /usr/lib64/atlas/liblapack.so.3
#3  0x00007fe44c058ad9 in dgelqf_ () from /usr/lib64/atlas/liblapack.so.3
#4  0x00007fe44c059a32 in dgels_ () from /usr/lib64/atlas/liblapack.so.3
#5  0x00007fe44f09fb3d in bool arma::auxlib::solve_ud<double, arma::Glue<arma::Mat<double>, arma::Mat<double>, arma::glue_times> >(arma::Mat<double>&, arma::Mat<double>&, arma::Base<double, arma::Glue<arma::Mat<double>, arma::Mat<double>, arma::glue_times> > const&) () at /usr/include/armadillo_bits/lapack_wrapper.hpp:677
#6  0x00007fe44f0a0f87 in arma::Col<double>::Col<arma::Glue<arma::Glue<arma::Mat<double>, arma::Mat<double>, arma::glue_times>, arma::Glue<arma::Mat<double>, arma::Mat<double>, arma::glue_times>, arma::glue_solve> >(arma::Base<double, arma::Glue<arma::Glue<arma::Mat<double>, arma::Mat<double>, arma::glue_times>, arma::Glue<arma::Mat<double>, arma::Mat<double>, arma::glue_times>, arma::glue_solve> > const&) ()
at /usr/include/armadillo_bits/glue_solve_meat.hpp:39

从堆栈跟踪中可以看出，我的 Armadillo 版本使用了 atlas。根据此文档，图集似乎是线程安全的:ftp://lsec.cc.ac.cn/netlib/atlas/faq.html#tsafe

2015 年 9 月 11 日更新

根据 Vladimir F 的建议，我终于有时间运行更多测试。

当我使用 ATLAS 的 BLAS 编译 Armadillo 时，我仍然能够重现然后挂起和 NaN。当它挂起时，堆栈跟踪中唯一改变的是对 BLAS 的调用:

#0  0x0000003fa8054718 in ATL_dscal_xp1yp0aXbX@plt () from /usr/lib64/atlas/libatlas.so.3
#1  0x0000003fb05e7666 in dlarfp_ () from /usr/lib64/atlas/liblapack.so.3
#2  0x0000003fb0576a61 in dgeqr2_ () from /usr/lib64/atlas/liblapack.so.3
#3  0x0000003fb0576e06 in dgeqrf_ () from /usr/lib64/atlas/liblapack.so.3
#4  0x0000003fb056d7d1 in dgels_ () from /usr/lib64/atlas/liblapack.so.3
#5  0x00007ff8f3de4c34 in void arma::lapack::gels<double>(char*, int*, int*, int*, double*, int*, double*, int*, double*, int*, int*) () at /usr/include/armadillo_bits/lapack_wrapper.hpp:677
#6  0x00007ff8f3de1787 in bool arma::auxlib::solve_od<double, arma::Glue<arma::Mat<double>, arma::Mat<double>, arma::glue_times> >(arma::Mat<double>&, arma::Mat<double>&, arma::Base<double, arma::Glue<arma::Mat<double>, arma::Mat<double>, arma::glue_times> > const&) () at /usr/include/armadillo_bits/auxlib_meat.hpp:3434

在没有 ATLAS 的情况下编译，仅使用 netlib BLAS 和 LAPACK，我能够重现 NaN，但不能重现挂起。

在这两种情况下，围绕 solve() 和 #pragma omp critical 我完全没有问题

最佳答案

您确定您的系统过度确定了吗？堆栈跟踪中的 solve_ud 另有说明。尽管您也有 solve_od ，但这可能与问题无关。但是，如果您认为系统应该不正常，找出发生这种情况的原因并加以修复并没有什么坏处。

Is armadillo solve() thread safe?

我认为这取决于您的 lapack 版本，另请参阅 this .看着code solve_od 所有访问的变量似乎都是本地的。注意代码中的警告:

NOTE: the dgels() function in the lapack library supplied by ATLAS 3.6 seems to have problems

因此，似乎只有 lapack::gels 会给您带来麻烦。如果无法修复 lapack，解决方法是堆叠您的系统并解决单个大型系统。如果您的个人系统很小，那可能会更有效。

关于c++ - Armadillo solve()线程安全吗？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/30063240/