@@ -469,10 +469,10 @@ X86_SIMD_SORT_INLINE type_t get_pivot_64bit(type_t *arr,
469
469
470
470
template <typename vtype, typename argtype, typename type_t >
471
471
X86_SIMD_SORT_INLINE void argsort_ (type_t *arr,
472
- arrsize_t *arg,
473
- arrsize_t left,
474
- arrsize_t right,
475
- arrsize_t max_iters)
472
+ arrsize_t *arg,
473
+ arrsize_t left,
474
+ arrsize_t right,
475
+ arrsize_t max_iters)
476
476
{
477
477
/*
478
478
* Resort to std::sort if quicksort isnt making any progress
@@ -498,17 +498,16 @@ X86_SIMD_SORT_INLINE void argsort_(type_t *arr,
498
498
argsort_<vtype, argtype>(
499
499
arr, arg, left, pivot_index - 1 , max_iters - 1 );
500
500
if (pivot != biggest)
501
- argsort_<vtype, argtype>(
502
- arr, arg, pivot_index, right, max_iters - 1 );
501
+ argsort_<vtype, argtype>(arr, arg, pivot_index, right, max_iters - 1 );
503
502
}
504
503
505
504
template <typename vtype, typename argtype, typename type_t >
506
505
X86_SIMD_SORT_INLINE void argselect_ (type_t *arr,
507
- arrsize_t *arg,
508
- arrsize_t pos,
509
- arrsize_t left,
510
- arrsize_t right,
511
- arrsize_t max_iters)
506
+ arrsize_t *arg,
507
+ arrsize_t pos,
508
+ arrsize_t left,
509
+ arrsize_t right,
510
+ arrsize_t max_iters)
512
511
{
513
512
/*
514
513
* Resort to std::sort if quicksort isnt making any progress
@@ -545,10 +544,10 @@ template <typename T,
545
544
template <typename ...>
546
545
typename half_vector>
547
546
X86_SIMD_SORT_INLINE void xss_argsort (T *arr,
548
- arrsize_t *arg,
549
- arrsize_t arrsize,
550
- bool hasnan = false ,
551
- bool descending = false )
547
+ arrsize_t *arg,
548
+ arrsize_t arrsize,
549
+ bool hasnan = false ,
550
+ bool descending = false )
552
551
{
553
552
/* TODO optimization: on 32-bit, use full_vector for 32-bit dtype */
554
553
using vectype = typename std::conditional<sizeof (T) == sizeof (int32_t ),
@@ -585,17 +584,19 @@ X86_SIMD_SORT_INLINE void avx512_argsort(T *arr,
585
584
bool hasnan = false ,
586
585
bool descending = false )
587
586
{
588
- xss_argsort<T, zmm_vector, ymm_vector>(arr, arg, arrsize, hasnan, descending);
587
+ xss_argsort<T, zmm_vector, ymm_vector>(
588
+ arr, arg, arrsize, hasnan, descending);
589
589
}
590
590
591
591
template <typename T>
592
592
X86_SIMD_SORT_INLINE void avx2_argsort (T *arr,
593
- arrsize_t *arg,
594
- arrsize_t arrsize,
595
- bool hasnan = false ,
596
- bool descending = false )
593
+ arrsize_t *arg,
594
+ arrsize_t arrsize,
595
+ bool hasnan = false ,
596
+ bool descending = false )
597
597
{
598
- xss_argsort<T, avx2_vector, avx2_half_vector>(arr, arg, arrsize, hasnan, descending);
598
+ xss_argsort<T, avx2_vector, avx2_half_vector>(
599
+ arr, arg, arrsize, hasnan, descending);
599
600
}
600
601
601
602
/* argselect methods for 32-bit and 64-bit dtypes */
@@ -605,10 +606,10 @@ template <typename T,
605
606
template <typename ...>
606
607
typename half_vector>
607
608
X86_SIMD_SORT_INLINE void xss_argselect (T *arr,
608
- arrsize_t *arg,
609
- arrsize_t k,
610
- arrsize_t arrsize,
611
- bool hasnan = false )
609
+ arrsize_t *arg,
610
+ arrsize_t k,
611
+ arrsize_t arrsize,
612
+ bool hasnan = false )
612
613
{
613
614
/* TODO optimization: on 32-bit, use full_vector for 32-bit dtype */
614
615
using vectype = typename std::conditional<sizeof (T) == sizeof (int32_t ),
@@ -635,10 +636,10 @@ X86_SIMD_SORT_INLINE void xss_argselect(T *arr,
635
636
636
637
template <typename T>
637
638
X86_SIMD_SORT_INLINE void avx512_argselect (T *arr,
638
- arrsize_t *arg,
639
- arrsize_t k,
640
- arrsize_t arrsize,
641
- bool hasnan = false )
639
+ arrsize_t *arg,
640
+ arrsize_t k,
641
+ arrsize_t arrsize,
642
+ bool hasnan = false )
642
643
{
643
644
xss_argselect<T, zmm_vector, ymm_vector>(arr, arg, k, arrsize, hasnan);
644
645
}
@@ -650,7 +651,8 @@ X86_SIMD_SORT_INLINE void avx2_argselect(T *arr,
650
651
arrsize_t arrsize,
651
652
bool hasnan = false )
652
653
{
653
- xss_argselect<T, avx2_vector, avx2_half_vector>(arr, arg, k, arrsize, hasnan);
654
+ xss_argselect<T, avx2_vector, avx2_half_vector>(
655
+ arr, arg, k, arrsize, hasnan);
654
656
}
655
657
656
658
#endif // XSS_COMMON_ARGSORT
0 commit comments