Skip to content

Commit 2c39de4

Browse files
author
Raghuveer Devulapalli
committed
Get rid of replace_inf_with_nan_fp16
1 parent 74734d2 commit 2c39de4

File tree

2 files changed

+3
-26
lines changed

2 files changed

+3
-26
lines changed

src/avx512-16bit-qsort.hpp

Lines changed: 1 addition & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -541,28 +541,6 @@ replace_nan_with_inf<zmm_vector<float16>>(uint16_t *arr, arrsize_t arrsize)
541541
return nan_count;
542542
}
543543

544-
X86_SIMD_SORT_INLINE_ONLY void replace_inf_with_nan_fp16(float16 *arr,
545-
arrsize_t size,
546-
arrsize_t nan_count,
547-
bool descending
548-
= false)
549-
{
550-
constexpr float16 quiet_NaN = {0x7c01};
551-
552-
if (descending) {
553-
for (arrsize_t ii = 0; nan_count > 0; ++ii) {
554-
arr[ii] = quiet_NaN;
555-
nan_count -= 1;
556-
}
557-
}
558-
else {
559-
for (arrsize_t ii = size - 1; nan_count > 0; --ii) {
560-
arr[ii] = quiet_NaN;
561-
nan_count -= 1;
562-
}
563-
}
564-
}
565-
566544
template <typename comparator>
567545
[[maybe_unused]] X86_SIMD_SORT_INLINE void
568546
avx512_qsort_fp16_helper(uint16_t *arr, arrsize_t arrsize)
@@ -623,8 +601,7 @@ avx512_qsort_fp16(uint16_t *arr,
623601
else {
624602
avx512_qsort_fp16_helper<Comparator<vtype, false>>(arr, arrsize);
625603
}
626-
replace_inf_with_nan_fp16(
627-
(float16 *)arr, arrsize, nan_count, descending);
604+
replace_inf_with_nan(arr, arrsize, nan_count, descending);
628605
}
629606

630607
#ifdef __MMX__

src/xss-common-qsort.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ X86_SIMD_SORT_INLINE void replace_inf_with_nan(type_t *arr,
116116
arr[ii] = xss::fp::quiet_NaN<type_t>();
117117
}
118118
else {
119-
arr[ii] = 0xFFFF;
119+
arr[ii] = 0x7c01; // std::quiet_nan
120120
}
121121
nan_count -= 1;
122122
}
@@ -127,7 +127,7 @@ X86_SIMD_SORT_INLINE void replace_inf_with_nan(type_t *arr,
127127
arr[ii] = xss::fp::quiet_NaN<type_t>();
128128
}
129129
else {
130-
arr[ii] = 0xFFFF;
130+
arr[ii] = 0x7c01; // std::quiet_nan
131131
}
132132
nan_count -= 1;
133133
}

0 commit comments

Comments
 (0)