File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
dpctl/tensor/libtensor/include/kernels/elementwise_functions Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -705,7 +705,9 @@ template <typename argTy1,
705705 typename T3,
706706 unsigned int vs,
707707 unsigned int nv>
708- class kernel_name >
708+ class kernel_name ,
709+ unsigned int vec_sz = 4 ,
710+ unsigned int n_vecs = 2 >
709711sycl::event binary_contig_impl (sycl::queue exec_q,
710712 size_t nelems,
711713 const char *arg1_p,
@@ -720,8 +722,6 @@ sycl::event binary_contig_impl(sycl::queue exec_q,
720722 cgh.depends_on (depends);
721723
722724 size_t lws = 64 ;
723- constexpr unsigned int vec_sz = 4 ;
724- constexpr unsigned int n_vecs = 2 ;
725725 const size_t n_groups =
726726 ((nelems + lws * n_vecs * vec_sz - 1 ) / (lws * n_vecs * vec_sz));
727727 const auto gws_range = sycl::range<1 >(n_groups * lws);
You can’t perform that action at this time.
0 commit comments