diff options
author | Raghuveer Devulapalli <raghuveer.devulapalli@intel.com> | 2022-09-27 21:58:29 -0700 |
---|---|---|
committer | Raghuveer Devulapalli <raghuveer.devulapalli@intel.com> | 2023-01-30 13:38:39 -0800 |
commit | ae978b8a2bc4e7b219d796519f9327feb08fe4e7 (patch) | |
tree | 8ee36270dbad3632820ad7ca126336a80803aafa /numpy/core/setup.py | |
parent | 49278b961b7254bc6a4aee478587c69682a3827e (diff) | |
download | numpy-ae978b8a2bc4e7b219d796519f9327feb08fe4e7.tar.gz |
ENH: Add AVX-512 based 64-bit dtype sort
Diffstat (limited to 'numpy/core/setup.py')
-rw-r--r-- | numpy/core/setup.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/numpy/core/setup.py b/numpy/core/setup.py index e509b9d11..912867709 100644 --- a/numpy/core/setup.py +++ b/numpy/core/setup.py @@ -650,6 +650,7 @@ def configuration(parent_package='',top_path=None): config.add_include_dirs(join('src', 'multiarray')) config.add_include_dirs(join('src', 'umath')) config.add_include_dirs(join('src', 'npysort')) + config.add_include_dirs(join('src', 'npysort', 'x86-simd-sort', 'src')) config.add_include_dirs(join('src', '_simd')) config.add_define_macros([("NPY_INTERNAL_BUILD", "1")]) # this macro indicates that Numpy build is in process @@ -942,7 +943,7 @@ def configuration(parent_package='',top_path=None): join('src', 'multiarray', 'usertypes.c'), join('src', 'multiarray', 'vdot.c'), join('src', 'common', 'npy_sort.h.src'), - join('src', 'npysort', 'x86-qsort.dispatch.cpp'), + join('src', 'npysort', 'x86-qsort-skx.dispatch.cpp'), join('src', 'npysort', 'quicksort.cpp'), join('src', 'npysort', 'mergesort.cpp'), join('src', 'npysort', 'timsort.cpp'), |