编译具有不同预处理器变量值的函数的多个版本

问：

我有以下函数，它使用 OpenMP 进行并行：

Real3 MyThing::compute(...) {
   // Do some setup...

#pragma omp parallel num_threads(OMP_THREADS)
    {
        // Local variables...
#pragma omp for nowait
        for (...) { ... }
#pragma omp critical
        { ... }
    }

    // Wrap up and return
}

其中使用的线程数可以在编译时由预处理器变量确定。OMP_THREADS

由于问题大小是在配置文件中确定的，因此仅在运行时才知道。因此，我想为 的多个值编译此函数（并且一个根本没有 OMP），然后有一个运行时条件来决定要运行哪个版本，例如：OMP_THREADS

if (mode == "ompx4") {
    return compute_omp4(...);
} else if (mode == "ompx8") {
    return compute_omp8(...);
} else {
    return compute_serial(...);
}

显然，我可以通过多次复制粘贴整个函数并在每个函数中硬编码不同的值来做到这一点，但由于这是一个相当大的函数，因此这并不理想。OMP_THREADS

通常我会对模板变量做一些类似的事情，但我认为我不能那样设置。关于如何以合理紧凑的形式制作它的任何建议？num_threads