@@ -161,24 +161,10 @@ topk_full_merge_sort_impl(sycl::queue &exec_q,
161
161
162
162
using IotaKernelName = topk_populate_index_data_krn<argTy, IndexTy, CompT>;
163
163
164
- #if 1
165
164
using dpctl::tensor::kernels::sort_utils_detail::iota_impl;
166
165
167
166
sycl::event populate_indexed_data_ev = iota_impl<IotaKernelName, IndexTy>(
168
167
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
182
168
183
169
std::size_t sorted_block_size;
184
170
// Sort segments of the array
@@ -513,23 +499,10 @@ sycl::event topk_radix_impl(sycl::queue &exec_q,
513
499
514
500
using IotaKernelName = topk_iota_krn<argTy, IndexTy>;
515
501
516
- #if 1
517
502
using dpctl::tensor::kernels::sort_utils_detail::iota_impl;
518
503
519
504
sycl::event iota_ev = iota_impl<IotaKernelName, IndexTy>(
520
505
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
533
506
534
507
sycl::event radix_sort_ev =
535
508
radix_sort_details::parallel_radix_sort_impl<IndexTy, IndexedProjT>(
0 commit comments