ENH, SIMD: dispatch argmax/armin for vx vxe

numpy · Feb 16, 2022 · 982ba4b · 982ba4b
1 parent aa81735
commit 982ba4b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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