Skip to content

Commit 48b15a0

Browse files
committed
Changed parameters for argsort/argselect
1 parent 758a25c commit 48b15a0

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/avx512-64bit-argsort.hpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -396,8 +396,8 @@ X86_SIMD_SORT_INLINE void argsort_64bit_(type_t *arr,
396396
/*
397397
* Base case: use bitonic networks to sort arrays <= 64
398398
*/
399-
if (right + 1 - left <= 64) {
400-
argsort_n<vtype, 64>(arr, arg + left, (int32_t)(right + 1 - left));
399+
if (right + 1 - left <= 256) {
400+
argsort_n<vtype, 256>(arr, arg + left, (int32_t)(right + 1 - left));
401401
return;
402402
}
403403
type_t pivot = get_pivot_64bit<vtype>(arr, arg, left, right);
@@ -429,8 +429,8 @@ X86_SIMD_SORT_INLINE void argselect_64bit_(type_t *arr,
429429
/*
430430
* Base case: use bitonic networks to sort arrays <= 64
431431
*/
432-
if (right + 1 - left <= 64) {
433-
argsort_n<vtype, 64>(arr, arg + left, (int32_t)(right + 1 - left));
432+
if (right + 1 - left <= 256) {
433+
argsort_n<vtype, 256>(arr, arg + left, (int32_t)(right + 1 - left));
434434
return;
435435
}
436436
type_t pivot = get_pivot_64bit<vtype>(arr, arg, left, right);

0 commit comments

Comments
 (0)