@@ -161,24 +161,10 @@ topk_full_merge_sort_impl(sycl::queue &exec_q,
161161
162162 using IotaKernelName = topk_populate_index_data_krn<argTy, IndexTy, CompT>;
163163
164- #if 1
165164 using dpctl::tensor::kernels::sort_utils_detail::iota_impl;
166165
167166 sycl::event populate_indexed_data_ev = iota_impl<IotaKernelName, IndexTy>(
168167 exec_q, index_data, iter_nelems * axis_nelems, depends);
169- #else
170- sycl::event populate_indexed_data_ev =
171- exec_q.submit([&](sycl::handler &cgh) {
172- cgh.depends_on(depends);
173-
174- auto const &range = sycl::range<1>(iter_nelems * axis_nelems);
175-
176- cgh.parallel_for<IotaKernelName>(range, [=](sycl::id<1> id) {
177- std::size_t i = id[0];
178- index_data[i] = static_cast<IndexTy>(i);
179- });
180- });
181- #endif
182168
183169 std::size_t sorted_block_size;
184170 // Sort segments of the array
@@ -513,23 +499,10 @@ sycl::event topk_radix_impl(sycl::queue &exec_q,
513499
514500 using IotaKernelName = topk_iota_krn<argTy, IndexTy>;
515501
516- #if 1
517502 using dpctl::tensor::kernels::sort_utils_detail::iota_impl;
518503
519504 sycl::event iota_ev = iota_impl<IotaKernelName, IndexTy>(
520505 exec_q, workspace, total_nelems, depends);
521- #else
522- sycl::event iota_ev = exec_q.submit([&](sycl::handler &cgh) {
523- cgh.depends_on(depends);
524-
525- cgh.parallel_for<IotaKernelName>(
526- sycl::range<1>(total_nelems), [=](sycl::id<1> id) {
527- size_t i = id[0];
528- IndexTy sort_id = static_cast<IndexTy>(i);
529- workspace[i] = sort_id;
530- });
531- });
532- #endif
533506
534507 sycl::event radix_sort_ev =
535508 radix_sort_details::parallel_radix_sort_impl<IndexTy, IndexedProjT>(
0 commit comments