diff options
author | Julian Taylor <juliantaylor108@gmail.com> | 2016-09-25 17:16:45 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2016-09-25 17:16:45 +0200 |
commit | 0887da9386e812d2769744afd6c58a927d6cbb45 (patch) | |
tree | 65f07b1cf5c4451be4e324e34fe8fd907578b07c /numpy/core/setup_common.py | |
parent | a17e905faa20fcdd1b3f039bceef6efa19671bb3 (diff) | |
parent | ae32e780d85e8e42402bc1fee2bbdb3ee90426c4 (diff) | |
download | numpy-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.py | 5 |
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 |