c - 用于并行的 OpenMP get_num_threads

Question

查找将用于 omp parallel for 的线程数的最佳方法是什么？我想分配足够的内存供所有线程使用，但在进入并行部分之前不能使用 omp_get_num_threads。例如:

int thread_count = ?
float *data = (float*)malloc(thread_count * data_count * sizeof(float));

#pragma omp parallel for default(shared) private(x)
for (x=0; x<N; x++)
{
    int threadid = omp_get_thread_num();
    //... do stuff with data
}

free(data);

这样做的正确方法是什么，以便我可以知道将使用多少线程并可以相应地进行分配？谢谢

Answer 1

您可以调用 omp_get_max_threads()。