如何使用fftw_malloc分配3D数组用于FFTW多线程3D DFT计算

改造方法 #

你不需要做3D数组降维映射，FFTW要求的输入是连续行优先存储的数组，fftw_malloc只是对分配的内存做了SIMD对齐优化，内存布局和你原来用new分配的3D数组完全一致，直接按以下步骤改造即可：

1. 3D数组分配改造 #

直接用fftw_malloc分配对应大小的连续内存，强转为你原来用的三维数组指针即可，原有数组访问逻辑V[i][j][k]不需要做任何修改：

const unsigned int RES = 256;
// 分配对齐的连续内存，强转为三维数组指针
std::complex<double>(*V)[RES][RES] = reinterpret_cast<std::complex<double>(*)[RES][RES]>(
    fftw_malloc(sizeof(std::complex<double>) * RES * RES * RES)
);

FFTW官方保证std::complex<double>和fftw_complex的内存布局完全兼容，这个类型转换是安全的。

2. Plan创建流程 #

这部分和你原有逻辑几乎一致，不需要额外修改：

int N_Threads = omp_get_max_threads();
fftw_init_threads();
fftw_plan_with_nthreads(N_Threads);

fftw_complex *input_V = reinterpret_cast<fftw_complex*>(V);
fftw_plan FORWARD_V = fftw_plan_dft_3d(RES, RES, RES, input_V, input_V, FFTW_FORWARD, FFTW_MEASURE);
fftw_plan BACKWARD_V = fftw_plan_dft_3d(RES, RES, RES, input_V, input_V, FFTW_BACKWARD, FFTW_MEASURE);

3. 资源释放注意事项 #

需要注意资源释放顺序不能乱，且不能用delete释放fftw_malloc分配的内存：

// 先销毁plan，再释放数组内存
fftw_destroy_plan(FORWARD_V);
fftw_destroy_plan(BACKWARD_V);
fftw_free(V);
// 程序退出前可选调用线程清理接口
fftw_cleanup_threads();

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