ENH: Vectorize np.partition and np.argpartition using AVX-512#24201

Merged

mattip merged 23 commits intonumpy:mainfrom

r-devulap:np-partition

Oct 1, 2023

Commits on Sep 18, 2023

Update submodule to latest
Raghuveer Devulapalli
committed
Add benchmarks for np.partition
Raghuveer Devulapalli
committed
ENH: Use avx512_qselect for np.partition
Raghuveer Devulapalli
committed
TST: Add tests for np.partition
Raghuveer Devulapalli
committed
BENCH: Add benchmarks for argpartition
Raghuveer Devulapalli
committed
Update x86-simd-sort submodule to latest
Raghuveer Devulapalli
committed
get_partition_func does not need an extra argument
Raghuveer Devulapalli
committed
ENH: Use avx512_argselect for np.argpartition
Raghuveer Devulapalli
committed
TST: add tests for np.argpartition
Raghuveer Devulapalli
committed
Enable AVX-512 partition and argpartition on Windows
Raghuveer Devulapalli
committed
Move PyArray_PartitionFunc and PyArray_ArgPartitionFunc out of ndarraytypes.h
Raghuveer Devulapalli
committed
MAINT: Prevent re-definition of quickselect dispatch on ARM 32-bit
Raghuveer Devulapalli
committed
Avoid using template specializations for quickselect and argquickselect dispatch
Raghuveer Devulapalli
committed
Update submodule to latest
Raghuveer Devulapalli
committed
Update x86-simd-sort submodule to latest
Raghuveer Devulapalli
committed
Update x86-simd-sort submodule to latest
Raghuveer Devulapalli
committed
ifdef remove avx512 instantiations for CYGWIN
Raghuveer Devulapalli
committed
Enable quicksort on WIN32
Raghuveer Devulapalli
committed
BUG: Fix compile error for avx512_qselect
Raghuveer Devulapalli
committed
BUG: Fix qselect function declaration argument
Raghuveer Devulapalli
committed
Disable avx512_qselect on 32-bit systems
Raghuveer Devulapalli
committed
Add may vary to output of np.partition and np.argpartition in docs
Raghuveer Devulapalli
committed

Commits on Sep 19, 2023

Use np::Half instead of np_tag::npy_half::type
Raghuveer Devulapalli
committed