summaryrefslogtreecommitdiff
path: root/numpy/core/setup_common.py
diff options
context:
space:
mode:
authorJulian Taylor <juliantaylor108@gmail.com>2016-09-25 17:16:45 +0200
committerGitHub <noreply@github.com>2016-09-25 17:16:45 +0200
commit0887da9386e812d2769744afd6c58a927d6cbb45 (patch)
tree65f07b1cf5c4451be4e324e34fe8fd907578b07c /numpy/core/setup_common.py
parenta17e905faa20fcdd1b3f039bceef6efa19671bb3 (diff)
parentae32e780d85e8e42402bc1fee2bbdb3ee90426c4 (diff)
downloadnumpy-0887da9386e812d2769744afd6c58a927d6cbb45.tar.gz
Merge pull request #7980 from juliantaylor/avx-runtime
ENH: Add ability to runtime select ufunc loops, add AVX2 integer loops
Diffstat (limited to 'numpy/core/setup_common.py')
-rw-r--r--numpy/core/setup_common.py5
1 files changed, 5 insertions, 0 deletions
diff --git a/numpy/core/setup_common.py b/numpy/core/setup_common.py
index 90b592999..cc2047b45 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,10 @@ OPTIONAL_FUNCTION_ATTRIBUTES = [('__attribute__((optimize("unroll-loops")))',
'attribute_optimize_opt_3'),
('__attribute__((nonnull (1)))',
'attribute_nonnull'),
+ ('__attribute__((target ("avx")))',
+ 'attribute_target_avx'),
+ ('__attribute__((target ("avx2")))',
+ 'attribute_target_avx2'),
]
# variable attributes tested via "int %s a" % attribute