ENH: Vectorize sort for int32 and uint32 using AVX-512

pmli · Jan 14, 2022 · f467d6b · f467d6b
1 parent 12d02f5
commit f467d6b
Show file tree

Hide file tree

Showing 4 changed files with 522 additions and 489 deletions.
diff --git a/numpy/core/setup.py b/numpy/core/setup.py
@@ -946,6 +946,7 @@ def gl_if_msvc(build_cmd):
             join('src', 'multiarray', 'usertypes.c'),
             join('src', 'multiarray', 'vdot.c'),
             join('src', 'common', 'npy_sort.h.src'),
+            join('src', 'npysort', 'qsort-32bit-avx512.h.src'),
             join('src', 'npysort', 'quicksort.c.src'),
             join('src', 'npysort', 'mergesort.c.src'),
             join('src', 'npysort', 'timsort.c.src'),