c - 如何避免分配到堆

Question

我试图通过不在循环中使用 malloc 来提高性能。由于数组的大小几乎总是相同的大小，因此我尝试使用堆栈变量并且仅在数组太大时才替换它。不幸的是，这似乎不是我得到的行为。这是我的代码片段。这里有什么明显的错误吗？

    double *A
    if(n>1024){
        A = malloc( n * sizeof( *A ) );
        if( !A ) {
               fprintf( stderr, "Failed to allocate phi in calculate_forces()\n" );
               exit( EXIT_FAILURE );
        }
    }
    else{
        double a[1024];
        A=a;
    }

Answer 1

你少了一个大括号，因此改变了实现的逻辑

double *A
if(n>1024){
    A = malloc( ng * sizeof( *phi ) );
    if( !phi ) {
           fprintf( stderr, "Failed to allocate phi in calculate_forces()\n" );
           exit( EXIT_FAILURE );
    }
}                 // this is the brace you need to add <<<<<<<<<<<<<<<<<
else{
    double a[1024];
    A=a;
}

编辑:

您的代码还有其他问题，例如将 a 分配给 A，其中 a 的生命周期比 A，无法判断您是否需要调用 free，使用 n 和 ng 而不是其中一个 - 可能是一个错误，并且可能还会混淆 phi 和 A