Skip to content

Commit 239aa78

Browse files
Enable radix sort for topk
1 parent 1cbed98 commit 239aa78

File tree

1 file changed

+1
-1
lines changed
  • dpctl/tensor/libtensor/source/sorting

1 file changed

+1
-1
lines changed

dpctl/tensor/libtensor/source/sorting/topk.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ sycl::event topk_caller(sycl::queue &exec_q,
114114
using dpctl::tensor::kernels::topk_radix_impl;
115115
const auto ascending = !largest;
116116

117-
if (axis_nelems > 16384) {
117+
if (true) {
118118
return topk_radix_impl<argTy, IndexTy>(
119119
exec_q, iter_nelems, axis_nelems, k, ascending, arg_cp, vals_cp,
120120
inds_cp, iter_arg_offset, iter_vals_offset, iter_inds_offset,

0 commit comments

Comments
 (0)