Use ChooseFunctor as choose kernel name

IntelPython · Jan 29, 2025 · ca14052 · ca14052
1 parent 620ae08
commit ca14052
Showing 1 changed file with 5 additions and 13 deletions.
diff --git a/dpnp/backend/extensions/indexing/choose_kernel.hpp b/dpnp/backend/extensions/indexing/choose_kernel.hpp
@@ -76,13 +76,6 @@ struct NthStrideOffsetUnpacked
 namespace dpnp::extensions::indexing::kernels
 {
 
-template <typename ProjectorT,
-          typename IndOutIndexerT,
-          typename ChoicesIndexerT,
-          typename T,
-          typename IndT>
-class choose_kernel;
-
 template <typename ProjectorT,
           typename IndOutIndexerT,
           typename ChoicesIndexerT,
@@ -176,14 +169,13 @@ sycl::event choose_impl(sycl::queue &q,
         const NthChoiceIndexerT choices_indexer{
             nd, chc_offsets, shape_and_strides, shape_and_strides + 3 * nd};
 
-        using KernelName = choose_kernel<ProjectorT, InOutIndexerT,
+        using ChooseFunc = ChooseFunctor<ProjectorT, InOutIndexerT,
                                          NthChoiceIndexerT, indTy, Ty>;
 
-        cgh.parallel_for<KernelName>(
-            sycl::range<1>(nelems),
-            ChooseFunctor<ProjectorT, InOutIndexerT, NthChoiceIndexerT, indTy,
-                          Ty>(ind_tp, dst_tp, chcs_cp, n_chcs, ind_out_indexer,
-                              choices_indexer));
+        cgh.parallel_for<ChooseFunc>(sycl::range<1>(nelems),
+                                     ChooseFunc(ind_tp, dst_tp, chcs_cp, n_chcs,
+                                                ind_out_indexer,
+                                                choices_indexer));
     });
 
     return choose_ev;