python - 通过python安装脚本以编程方式测试对openmp的支持

我正在使用cython和c加快时间敏感操作的python项目。在一些cython例程中，如果空闲内核可用，我们将使用openmp进一步加快操作速度。

由于最新OS版本的默认编译器(10.7和10.8上的llvm/clang)不支持openmp，因此这在OS X上引起了一些烦人的情况。我们的权宜之计解决方案是告诉人们在构建时将gcc设置为其编译器。我们真的很想以编程方式执行此操作，因为clang可以毫无问题地构建其他所有内容。

现在，编译将失败，并显示以下错误:

clang: error: linker command failed with exit code 1 (use -v to see invocation)
error: Command "cc -bundle -undefined dynamic_lookup -L/usr/local/lib -L/usr/local/opt/sqlite/lib build/temp.macosx-10.8-x86_64-2.7/yt/utilities/lib/geometry_utils.o -lm -o yt/utilities/lib/geometry_utils.so -fopenmp" failed with exit status 1

我们的安装脚本的相关部分如下所示:

config.add_extension("geometry_utils",
        ["yt/utilities/lib/geometry_utils.pyx"],
         extra_compile_args=['-fopenmp'],
         extra_link_args=['-fopenmp'],
         libraries=["m"], depends=["yt/utilities/lib/fp_utils.pxd"])

完整的setup.py文件是here。

有没有一种方法可以从安装脚本中以编程方式测试对openmp的支持？

最佳答案

通过检查测试程序是否可以编译，我能够使它工作:

import os, tempfile, subprocess, shutil

# see http://openmp.org/wp/openmp-compilers/
omp_test = \
r"""
#include <omp.h>
#include <stdio.h>
int main() {
#pragma omp parallel
printf("Hello from thread %d, nthreads %d\n", omp_get_thread_num(), omp_get_num_threads());
}
"""

def check_for_openmp():
    tmpdir = tempfile.mkdtemp()
    curdir = os.getcwd()
    os.chdir(tmpdir)

    filename = r'test.c'
    with open(filename, 'w', 0) as file:
        file.write(omp_test)
    with open(os.devnull, 'w') as fnull:
        result = subprocess.call(['cc', '-fopenmp', filename],
                                 stdout=fnull, stderr=fnull)

    os.chdir(curdir)
    #clean up
    shutil.rmtree(tmpdir)

    return result

关于python - 通过python安装脚本以编程方式测试对openmp的支持，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/16549893/