Cython 条件编译基于通过“setuptools”给出的外部值

Question

我尝试从 Cython pyx 文件有条件地生成 C 代码。我在 Cython 文档中发现，我可以使用 DEF 定义一个值，并使用 IF 根据定义的值有条件地生成代码，但是我如何从 @987654326 通过 Extension 设置来自 setup.py 的值@。

谢谢

Answer 1

感谢您的链接。

setup.py 中有趣的标志是cython_compile_time_env。并从 Cython 导入 Extension。

from setuptools import setup
from Cython.Distutils.extension import Extension

ext = Extension(
    name,
    include_dirs=include_dirs,
    cython_compile_time_env=dict(OPENMP=True),
    sources=['test.pyx'])

setup(name=name,
      cmdclass=dict(build_ext=build_ext),
      ext_modules=[ext])

而在test.pyx：

...
IF OPENMP:
#Do openmp
ELSE:
#No openmp
...

Cython 条件语句（上面的IF...ELSE）记录在here。