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

Question

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

谢谢

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。