|
17 | 17 | namespace np { namespace qsort_simd {
|
18 | 18 |
|
19 | 19 | #if defined(NPY_HAVE_AVX512_SKX)
|
20 |
| -template<> void NPY_CPU_DISPATCH_CURFX(ArgQSelect)(int32_t *arr, npy_intp* arg, npy_intp num, npy_intp kth) |
21 |
| -{ |
22 |
| - avx512_argselect(arr, reinterpret_cast<int64_t*>(arg), kth, num); |
23 |
| -} |
24 |
| -template<> void NPY_CPU_DISPATCH_CURFX(ArgQSelect)(uint32_t *arr, npy_intp* arg, npy_intp num, npy_intp kth) |
25 |
| -{ |
26 |
| - avx512_argselect(arr, reinterpret_cast<int64_t*>(arg), kth, num); |
27 |
| -} |
28 |
| -template<> void NPY_CPU_DISPATCH_CURFX(ArgQSelect)(int64_t*arr, npy_intp* arg, npy_intp num, npy_intp kth) |
29 |
| -{ |
30 |
| - avx512_argselect(arr, reinterpret_cast<int64_t*>(arg), kth, num); |
31 |
| -} |
32 |
| -template<> void NPY_CPU_DISPATCH_CURFX(ArgQSelect)(uint64_t*arr, npy_intp* arg, npy_intp num, npy_intp kth) |
33 |
| -{ |
34 |
| - avx512_argselect(arr, reinterpret_cast<int64_t*>(arg), kth, num); |
35 |
| -} |
36 |
| -template<> void NPY_CPU_DISPATCH_CURFX(ArgQSelect)(float *arr, npy_intp* arg, npy_intp num, npy_intp kth) |
37 |
| -{ |
38 |
| - avx512_argselect(arr, reinterpret_cast<int64_t*>(arg), kth, num); |
39 |
| -} |
40 |
| -template<> void NPY_CPU_DISPATCH_CURFX(ArgQSelect)(double *arr, npy_intp* arg, npy_intp num, npy_intp kth) |
41 |
| -{ |
42 |
| - avx512_argselect(arr, reinterpret_cast<int64_t*>(arg), kth, num); |
43 |
| -} |
44 | 20 | template<> void NPY_CPU_DISPATCH_CURFX(QSelect)(int32_t *arr, npy_intp num, npy_intp kth)
|
45 | 21 | {
|
46 | 22 | avx512_qselect(arr, kth, num, true);
|
|
0 commit comments