Skip to content

Commit a4bc0c4

Browse files
committed
Tweak set_bucket_id for boolean case
1 parent 639ad23 commit a4bc0c4

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

dpctl/tensor/libtensor/include/kernels/sorting/radix_common.hpp

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -345,7 +345,12 @@ T set_bucket_id(T val, T insert, std::uint32_t radix_offset)
345345
insert &= m;
346346
insert <<= radix_offset;
347347
m <<= radix_offset;
348-
return (val & ~m) | insert;
348+
if constexpr (std::is_same_v<T, bool>) {
349+
return (val & !m) | insert;
350+
}
351+
else {
352+
return (val & ~m) | insert;
353+
}
349354
}
350355

351356
} // namespace radix_common

0 commit comments

Comments
 (0)