Fixed data allocation

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() {