启用ARMA_USE_FFTW3时Armadillo代码与FFTW切片操作的线程安全性咨询

你的示例代码不具备线程安全性，核心问题出在FFTW规划器的全局竞争上，具体细节和解决方案如下：

1. FFTW规划器的线程安全问题 #

FFTW官方明确指出，其规划器（planner）并非线程安全——多个线程同时触发规划操作时，会竞争访问FFTW的全局状态，导致未定义行为（比如程序崩溃、计算结果错误）。

Armadillo默认情况下没有对FFTW规划器的线程安全做封装处理：当你在多线程中调用arma::fft()时，第一次针对特定尺寸的FFT调用会触发FFTW的规划流程，此时多个线程同时执行规划，必然引发竞争。

2. 切片FFTW操作的线程安全 #

当FFTW完成某一尺寸的规划并缓存计划后，后续基于该计划的实际FFT计算是线程安全的——只要每个线程使用独立的数据缓冲区（你的示例中每个线程处理独立的切片，数据缓冲区是独立的，这部分没问题）。但问题的关键在于，规划阶段的全局竞争无法避免，除非提前做处理。

解决方案 #

方案一：提前预规划FFT尺寸 #

在主线程启动所有工作线程前，先对后续需要用到的FFT尺寸执行一次arma::fft()，让FFTW完成规划并缓存计划。后续线程调用时会直接使用已缓存的计划，不会触发规划操作，避免竞争：

int main()
{
    // 预规划和线程中y相同尺寸的FFT（这里y是128长度的vec）
    vec dummy(128);
    arma::fft(dummy);
    
    // 原代码的线程创建、执行逻辑不变...
}

方案二：给规划操作加全局互斥锁 #

在每个线程调用arma::fft()时，通过全局互斥锁确保同一时间只有一个线程进入规划阶段，避免竞争：

#include <mutex>
std::mutex fftw_planner_mutex;

void threadFunction(const mat& dataSlice, int index, vector<float>& results)
{
    vec y = mean(dataSlice, 1);
    cx_vec a;
    
    // 锁定互斥锁，确保规划操作串行执行
    {
        std::lock_guard<std::mutex> lock(fftw_planner_mutex);
        a = arma::fft(y);
    }
    
    double amax = arma::max(arma::abs(a));
    results[index] = amax;
}

方案三：启用FFTW的线程安全模式 #

调用FFTW的线程初始化函数，开启线程安全支持：

#include <fftw3.h>

int main()
{
    // 初始化FFTW线程支持
    fftw_init_threads();
    // 设置FFTW使用的线程数（可选）
    fftw_plan_with_nthreads(std::thread::hardware_concurrency());
    
    // 原代码的线程创建、执行逻辑不变...
    
    // 程序结束时清理FFTW线程资源
    fftw_cleanup_threads();
}

注意：这种方式仅能让FFTW的规划器在多线程环境下更安全，但仍建议配合预规划使用，避免潜在的竞争风险。

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