python - numpy 的基本操作是否矢量化，即它们是否使用 SIMD 操作？

Question

我正在做一些性能分析，我想知道，当数据类型已知( double )时，numpy 是否对其标准数组操作进行矢量化。

a, b = (some numpy arrays)
c = a + b #Is this vectorized?

编辑:此操作是否矢量化，即计算是否包含 SIMD 操作？

Answer 1

是的，他们是。

/*
 * This file is for the definitions of simd vectorized operations.
 *
 * Currently contains sse2 functions that are built on amd64, x32 or
 * non-generic builds (CFLAGS=-march=...)
 * In future it may contain other instruction sets like AVX or NEON     detected
 * at runtime in which case it needs to be included indirectly via a file
 * compiled with special options (or use gcc target attributes) so the binary
 * stays portable.
 */

链接:Numpy simd.inc.src on github.