Skip to content

Commit 2e42c4b

Browse files
author
Raghuveer Devulapalli
committed
MAINT: Fix up arguments of avx512_*_fp16 functions
1 parent 51c0d52 commit 2e42c4b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

numpy/_core/src/npysort/x86_simd_qsort_16bit.dispatch.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ template<> void NPY_CPU_DISPATCH_CURFX(QSelect)(Half *arr, npy_intp num, npy_int
1313
#if defined(NPY_HAVE_AVX512_SPR)
1414
x86simdsortStatic::qselect(reinterpret_cast<_Float16*>(arr), kth, num, true);
1515
#else
16-
avx512_qselect_fp16(reinterpret_cast<uint16_t*>(arr), kth, num, true);
16+
avx512_qselect_fp16(reinterpret_cast<uint16_t*>(arr), kth, num, true, false);
1717
#endif
1818
}
1919

@@ -35,7 +35,7 @@ template<> void NPY_CPU_DISPATCH_CURFX(QSort)(Half *arr, npy_intp size)
3535
#if defined(NPY_HAVE_AVX512_SPR)
3636
x86simdsortStatic::qsort(reinterpret_cast<_Float16*>(arr), size, true);
3737
#else
38-
avx512_qsort_fp16(reinterpret_cast<uint16_t*>(arr), size, true);
38+
avx512_qsort_fp16(reinterpret_cast<uint16_t*>(arr), size, true, false);
3939
#endif
4040
}
4141
template<> void NPY_CPU_DISPATCH_CURFX(QSort)(uint16_t *arr, npy_intp size)

0 commit comments

Comments
 (0)