Skip to content

Pythran 作为 Numpy 后端

原文： http://docs.cython.org/en/latest/src/userguide/numpy_pythran.html

使用标志--np-pythran，可以将 Pythran numpy 实现用于与 numpy 相关的操作。使用此后端的一个优点是 Pythran 实现使用 C ++表达式模板来节省内存传输，并且可以受益于现代 CPU 的 SIMD 指令。

在某些情况下，这可以带来非常有趣的加速，从 2 到 16，具体取决于目标 CPU 架构和原始算法。

请注意，此功能是实验性的。

使用 setuptools 的用法示例

你首先需要安装 Pythran。有关更多信息，请参见其文档。

然后，只需在 Python 文件的顶部添加一个cython: np_pythran=True指令，该指令需要使用 Pythran numpy 支持进行编译。

以下是使用 setuptools 的简单setup.py文件的示例：

from setuptools import setup
from Cython.Build import cythonize
import numpy
import pythran

setup(
    name = "My hello app",
    ext_modules = cythonize('hello_pythran.pyx'),
    include_dirs = [numpy.get_include(), pythran.get_include()]
)

然后，使用hello_pythran.pyx中的以下标题：

# cython: np_pythran=True

hello_pythran.pyx将使用 Pythran numpy 支持编译。

请注意，可以通过在$HOME/.pythranrc文件中添加设置来进一步调整 Pythran。例如，这可以用于启用 Boost.SIMD 支持。有关更多信息，请参见 Pythran 用户手册。

回到顶部