Armadillo库eigs_gen函数多线程运行异常原因及可行性问询

核心问题 #

在C++ Armadillo库中，调用求解复稀疏矩阵特征值的eigs_gen函数时，即使线程间无任何数据共享，多线程并发运行仍会出现崩溃、返回错误结果等异常。需明确问题原因及该函数是否支持安全的多线程调用。

最小复现示例 #

以下代码在两个独立线程中分别求解大型复方阵的最小特征值及特征向量，线程间无数据共享，但仅在eigs_gen调用前后加互斥锁时才能正常运行：

#include<iostream>
#include<complex>
#include<exception>
#include <armadillo>
#include<thread>
#include<string>

using namespace std;
using namespace arma;

std::mutex stupid_lock;

void get_eigenvalue(int id, size_t N, double* out)
{
    sp_cx_mat H (N, N); // 稀疏矩阵，默认初始化为全0

    // 构造厄米矩阵（理论上特征值应为实数）
    for (uint i = 0; i <N; ++i)
    {
        H(i, i)=1*i;

        if (i>0)
            H(i, i-1)=-1i*double(i);
        if (i<N-1)
            H(i, i+1)=+1i*double(i);
    }

    cx_vec eigval;
    cx_mat eigvec;

    stupid_lock.lock();
    eigs_gen( eigval, eigvec, H, 1, "sr"); // 求解最小特征值
    stupid_lock.unlock();

    *out = eigval[0].real();

    // 合法性检查：厄米矩阵特征值虚部应接近0
    if ( eigval[0].imag()> 1e-9 )
    {
        throw std::runtime_error("线程"+to_string(id)+"错误：特征值为"+to_string(eigval[0].real())+"+"+to_string(eigval[0].imag())+"*i，虚部非零");
    }
}

int main()
{
    size_t N = 20000; // 矩阵规模足够大时才会触发异常
    double out1=0;
    double out2=0;

    thread t1(get_eigenvalue,1,N,&out1);
    thread t2(get_eigenvalue,2,N,&out2);

    t1.join();
    t2.join();
    cout<<"两个结果应一致："<<out1<<" "<<out2<<endl;

    return 0;
}

运行结果 #

• 加互斥锁时，程序稳定运行，输出符合预期：

$ bin/minimum_failure.exe
两个结果应一致：-19936.6 -19936.6

• 移除互斥锁后，出现多种异常：
  • 段错误（core dumped）
  • 抛出ARPACK错误并崩溃
  • 返回随机错误的实特征值
  • 厄米矩阵返回复特征值（违背数学原理）

编译环境与系统信息 #

• 发行版：Arch Linux（Linux 6.1.15-1-lts）
• Armadillo版本：12.0.1-1
• OpenBLAS版本：0.3.21-4（提供BLAS 3.9.0）
• GCC版本：12.2.1-2
• 编译参数：-std=c++17 -O2 -larmadillo -lpthread
• CPU线程数：20

问题原因与解答 #

原因分析 #

问题的核心在于eigs_gen的底层依赖库ARPACK本身不是线程安全的：

1. ARPACK内部使用全局变量存储计算过程中的工作区、状态信息等数据，当多个线程同时调用ARPACK接口时，会出现全局资源的竞争覆盖，直接导致内存越界、计算逻辑混乱。
2. 即使线程间的矩阵数据完全独立，ARPACK的全局状态被多线程篡改后，会引发计算错误甚至崩溃，这和观察到的现象完全匹配。

另外，OpenBLAS的多线程调度如果和ARPACK的全局状态冲突，会进一步加剧问题，但根本原因还是ARPACK的非线程安全特性。

能否安全多线程运行？ #

默认情况下不能直接多线程并发调用eigs_gen。若要实现安全的多线程计算，有两种可行方案：

1. 串行化ARPACK调用：像示例中那样，用互斥锁将eigs_gen的调用串行化，避免多个线程同时访问ARPACK的全局资源。
2. 使用线程安全的ARPACK版本：部分Linux发行版提供了线程安全的ARPACK包（比如arpack-threads），或者手动编译ARPACK时启用线程安全选项（通过-DARPACK_THREADSAFE等编译宏），此时Armadillo调用eigs_gen时可安全并发。

内容的提问来源于stack exchange，提问作者Nikolaj