File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed
dpctl/tensor/libtensor/include/kernels Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change 3131#include " dpctl_tensor_types.hpp"
3232#include " kernels/alignment.hpp"
3333#include " utils/offset_utils.hpp"
34+ #include " utils/sycl_utils.hpp"
3435#include " utils/type_utils.hpp"
3536
3637namespace dpctl
@@ -42,6 +43,8 @@ namespace kernels
4243namespace copy_as_contig
4344{
4445
46+ using dpctl::tensor::sycl_utils::sub_group_store;
47+
4548template <typename T,
4649 typename IndexerT,
4750 std::uint8_t vec_sz = 4u ,
@@ -113,7 +116,7 @@ class CopyAsCContigFunctor
113116 const ssize_t src_offset = src_indexer (elem_id);
114117 dst_vec[k] = src_p[src_offset];
115118 }
116- sg. store <vec_sz>(dst_multi_ptr , dst_vec);
119+ sub_group_store <vec_sz>(sg , dst_vec, dst_multi_ptr );
117120 }
118121 }
119122 else {
You can’t perform that action at this time.
0 commit comments