diff options
Diffstat (limited to 'modules/remote-sycl/c++/data.hpp')
| -rw-r--r-- | modules/remote-sycl/c++/data.hpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/modules/remote-sycl/c++/data.hpp b/modules/remote-sycl/c++/data.hpp index c2ee098..4aec046 100644 --- a/modules/remote-sycl/c++/data.hpp +++ b/modules/remote-sycl/c++/data.hpp @@ -16,6 +16,7 @@ public: private: // cl::sycl::buffer<data<Sch, encode::Native>> data_; using sycl_usm_allocator = acpp::sycl::usm_allocator<data<Sch,Encode>, acpp::sycl::usm::alloc::shared>; + sycl_usm_allocator sycl_alloc_; data<schema::FixedArray<schema::UInt64, Dim>, Encode> dims_; data<schema::UInt64, Encode> size_; std::vector<data<Sch,Encode>, sycl_usm_allocator> data_; @@ -32,9 +33,10 @@ private: } public: data(acpp::sycl::queue& q__): + sycl_alloc_{q__}, dims_{}, size_{0u}, - data_{0u,q__} + data_{0u,sycl_alloc_} { for(uint64_t iter = 0; iter < Dim; ++iter){ dims_.at({iter}) = 0u; @@ -42,9 +44,10 @@ public: } data(const data<schema::FixedArray<schema::UInt64, Dim>, Encode>& dims__, acpp::sycl::queue& q__): + sycl_alloc_{q__}, dims_{dims__}, size_{get_full_size()}, - data_{size_,q__} + data_{size_,sycl_alloc_} {} auto* get_internal_data() { |