Skip to content

Commit 5191d68

Browse files
committed
Triage topk radix sort test failure by disabling one-work-group radix kernels
1 parent 58da416 commit 5191d68

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

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

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1483,7 +1483,10 @@ sycl::event parallel_radix_sort_impl(sycl::queue &exec_q,
14831483
dev.template get_info<sycl::info::device::max_work_group_size>();
14841484

14851485
constexpr std::uint16_t ref_wg_size = 64;
1486-
if (n_to_sort <= 16384 && ref_wg_size * 8 <= max_wg_size) {
1486+
constexpr bool enable_one_wg_radix_sort = false;
1487+
if (enable_one_wg_radix_sort && n_to_sort <= 16384 &&
1488+
ref_wg_size * 8 <= max_wg_size)
1489+
{
14871490
using _RadixSortKernel = OneWorkGroupRadixSortKernel<ValueT, ProjT>;
14881491

14891492
if (n_to_sort <= 64 && ref_wg_size <= max_wg_size) {

0 commit comments

Comments
 (0)