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,
705
705
typename T3,
706
706
unsigned int vs,
707
707
unsigned int nv>
708
- class kernel_name >
708
+ class kernel_name ,
709
+ unsigned int vec_sz = 4 ,
710
+ unsigned int n_vecs = 2 >
709
711
sycl::event binary_contig_impl (sycl::queue exec_q,
710
712
size_t nelems,
711
713
const char *arg1_p,
@@ -720,8 +722,6 @@ sycl::event binary_contig_impl(sycl::queue exec_q,
720
722
cgh.depends_on (depends);
721
723
722
724
size_t lws = 64 ;
723
- constexpr unsigned int vec_sz = 4 ;
724
- constexpr unsigned int n_vecs = 2 ;
725
725
const size_t n_groups =
726
726
((nelems + lws * n_vecs * vec_sz - 1 ) / (lws * n_vecs * vec_sz));
727
727
const auto gws_range = sycl::range<1 >(n_groups * lws);
You can’t perform that action at this time.
0 commit comments