summaryrefslogtreecommitdiff
path: root/numpy/core/setup_common.py
diff options
context:
space:
mode:
authorJulian Taylor <jtaylor.debian@googlemail.com>2016-08-27 10:21:40 +0200
committerJulian Taylor <jtaylor.debian@googlemail.com>2016-08-27 10:36:37 +0200
commit10723a9b73feb59278bf6e0c8ad0830c3ddb0600 (patch)
treee8f422c138e641cf89c48bd038cd3a064c542de3 /numpy/core/setup_common.py
parenteeb4e17a165e90430a01936914afb2bbeb34acc7 (diff)
downloadnumpy-10723a9b73feb59278bf6e0c8ad0830c3ddb0600.tar.gz
MAINT: add avx __builtin_cpu_supports and target attribute checks
NPY_CPU_SUPPORTS_AVX2 checks at runtime if AVX2 is supported
Diffstat (limited to 'numpy/core/setup_common.py')
-rw-r--r--numpy/core/setup_common.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/numpy/core/setup_common.py b/numpy/core/setup_common.py
index ba7521e30..2a42a0638 100644
--- a/numpy/core/setup_common.py
+++ b/numpy/core/setup_common.py
@@ -125,6 +125,7 @@ OPTIONAL_INTRINSICS = [("__builtin_isnan", '5.'),
("__builtin_bswap64", '5u'),
("__builtin_expect", '5, 0'),
("__builtin_mul_overflow", '5, 5, (int*)5'),
+ ("__builtin_cpu_supports", '"sse"'),
("_mm_load_ps", '(float*)0', "xmmintrin.h"), # SSE
("_mm_prefetch", '(float*)0, _MM_HINT_NTA',
"xmmintrin.h"), # SSE
@@ -141,6 +142,8 @@ OPTIONAL_FUNCTION_ATTRIBUTES = [('__attribute__((optimize("unroll-loops")))',
'attribute_optimize_opt_3'),
('__attribute__((nonnull (1)))',
'attribute_nonnull'),
+ ('__attribute__((target ("avx2")))',
+ 'attribute_target_avx2'),
]
# variable attributes tested via "int %s a" % attribute