Skip to content

Commit 9810e05

Browse files
author
Raghuveer Devulapalli
committed
Explicit instantiation of avx512_partial_qsort for _Float16
1 parent 3dab097 commit 9810e05

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

src/avx512fp16-16bit-qsort.hpp

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -201,4 +201,13 @@ void avx512_qselect(_Float16 *arr, arrsize_t k, arrsize_t arrsize, bool hasnan)
201201
arr, k, 0, indx_last_elem, 2 * (arrsize_t)log2(indx_last_elem));
202202
}
203203
}
204+
template <>
205+
void avx512_partial_qsort(_Float16 *arr,
206+
arrsize_t k,
207+
arrsize_t arrsize,
208+
bool hasnan)
209+
{
210+
avx512_qselect(arr, k - 1, arrsize, hasnan);
211+
avx512_qsort(arr, k - 1);
212+
}
204213
#endif // AVX512FP16_QSORT_16BIT

0 commit comments

Comments
 (0)