Cython ipython魔术与编译时环境变量

时间:2017-01-07 21:04:33

标签: python ipython cython ipython-magic

%%cython命令非常方便创建cython函数而无需构建和使用包。该命令有几个选项,但我找不到在那里指定编译时环境变量的方法。

我想要相当于:

from Cython.Distutils.extension import Extension
ext = Extension(...
                cython_compile_time_env={'MYVAR': 10},
                ...)

表示%%cython命令。

我已经尝试过:

%%cython -cython_compile_time_env={'MYVAR':10}

IF MYVAR:
    def func():
        return 1
ELSE:
    def func():
        return 2

但是这引发了一个例外:

Error compiling Cython file:
------------------------------------------------------------
...

IF MYVAR:
       ^
------------------------------------------------------------

...\.ipython\cython\_cython_magic_28df41ea67fec254f0be4fc74f7a6a54.pyx:2:8: Compile-time name 'MYVAR' not defined

%%cython --cython_compile_time_env={'MYVAR':10}

IF MYVAR:
    def func():
        return 1
ELSE:
    def func():
        return 2

引发

  

UsageError:无法识别的参数: - cython_compile_time_env = {' MYVAR':10}

1 个答案:

答案 0 :(得分:1)

这只是一种解决方法,而不是适当的解决方案,但是可以实现所需的性能。简而言之,无法通过compile_time_env魔术来提供%%cython,但是对cythonize的调用将采用可以直接修改的默认编译器选项。对于上面的示例,请尝试以下操作。

from Cython.Compiler.Main import default_options

default_options['compile_time_env'] = {'MYVAR': 0}
# Running the magic and calling `func` yields 2

default_options['compile_time_env'] = {'MYVAR': 1}
# Running the magic and calling `func` yields 1