OpenMP并行循环中malloc/free的最优实现方式咨询

问题背景 #

在百万次迭代的循环中，原串行实现每次迭代都调用malloc和free，已知这种频繁的内存分配释放会造成内存碎片。现在有两种OpenMP并行化方案，需要对比优劣。

原串行循环代码 #

for (size_t i = 0 ; i < 1000000; ++i){
    double * p = malloc(sizeof(double)*FIXED_SIZE); 

    /* FIXED_SIZE是整个循环的常量，但仅能动态确定 */

    /* 使用p数组执行逻辑操作 */

    free(p);
}

方案一：并行循环内加atomic保护malloc/free #

#pragma omp parallel for
for (size_t i = 0 ; i < 1000000; ++i){

    #pragma omp atomic
    double * p = malloc(sizeof(double)*FIXED_SIZE); 
    
    /* 使用p数组执行逻辑操作 */

    #pragma omp atomic
    free(p);
}

方案一的问题 #

1. 错误的atomic使用：malloc和free本身是线程安全的（多数现代标准库实现），不需要用#pragma omp atomic保护。这个指令会强制malloc/free操作串行执行，彻底抵消并行循环的性能优势，甚至比串行版本更慢——因为多了同步开销。
2. 未解决内存碎片：每次迭代依然会分配和释放内存，百万次调用依然会产生大量内存碎片，没有解决核心问题。

方案二：预分配线程私有共享数组 #

// 修正原代码潜在问题：需在并行区内获取实际线程数
double *p = NULL;
int num_threads;

#pragma omp parallel
{
    #pragma omp single
    {
        num_threads = omp_get_num_threads();
        p = malloc(sizeof(double) * FIXED_SIZE * num_threads);
    }
}

#pragma omp parallel for
for (size_t i = 0 ; i < 1000000; ++i){
    int thread_num = omp_get_thread_num();
    double *p1 = p + FIXED_SIZE * thread_num ;
    
    /* 使用p1数组执行逻辑操作 */
}
free(p);

方案二的优势 #

1. 消除频繁内存操作：仅在循环前分配一次内存，循环后释放一次，完全消除了百万次malloc/free的开销，从根源上避免了内存碎片。
2. 无同步开销：每个线程使用预先分配的独立内存块，线程间无数据竞争，不需要任何同步指令，并行效率最高。
3. 缓存友好：线程固定复用自己的内存块，缓存命中率远高于每次分配新内存，进一步提升性能。

结论 #

方案二是完全优于方案一的正确实现。方案一不仅没有解决内存碎片问题，还因错误使用同步指令导致并行性能严重退化；方案二则通过预分配线程私有内存，完美解决了内存碎片和频繁分配释放的性能问题，是这类场景下的标准优化手段。

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