diff options
| -rw-r--r-- | numpy/core/src/multiarray/argfunc.dispatch.c.src | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/numpy/core/src/multiarray/argfunc.dispatch.c.src b/numpy/core/src/multiarray/argfunc.dispatch.c.src index cbfaebdb4..1d7753275 100644 --- a/numpy/core/src/multiarray/argfunc.dispatch.c.src +++ b/numpy/core/src/multiarray/argfunc.dispatch.c.src @@ -4,6 +4,7 @@ ** sse2 sse42 xop avx2 avx512_skx ** vsx2 ** neon asimd + ** vx vxe **/ #define NPY_NO_DEPRECATED_API NPY_API_VERSION @@ -123,7 +124,7 @@ simd_@func@_@sfx@(npyv_lanetype_@sfx@ *ip, npy_intp len) * #bsfx = b32, b32, b64, b64, b32, b64# * #is_fp = 0*4, 1*2# * #is_idx32 = 1*2, 0*2, 1, 0# - * #chk_simd = NPY_SIMD*5, NPY_SIMD_F64# + * #chk_simd = NPY_SIMD*4, NPY_SIMD_F32, NPY_SIMD_F64# */ #if @chk_simd@ /**begin repeat1 @@ -298,6 +299,9 @@ scalar_loop: #if NPY_BITSOF_@BTYPE@ == 64 && !NPY_SIMD_F64 #undef TO_SIMD_SFX #endif + #if NPY_BITSOF_@BTYPE@ == 32 && !NPY_SIMD_F32 + #undef TO_SIMD_SFX + #endif #elif @is_unsigned@ #define TO_SIMD_SFX(X) X##_u@len@ #else |