Preparing benchmark work

1 files changed, 7 insertions, 1 deletions

diff --git a/modules/remote-sycl/c++/data.hpp b/modules/remote-sycl/c++/data.hpp
index e436e72..d939a53 100644
--- a/modules/remote-sycl/c++/data.hpp
+++ b/modules/remote-sycl/c++/data.hpp
@@ -12,9 +12,11 @@ template<typename Schema>
 class data<Schema, encode::Native, rmt::Sycl> {
 private:
 	cl::sycl::buffer<data<Schema, encode::Native, storage::Default>> data_;
+	uint64_t size_;
 public:
 	data(const data<Schema, encode::Native, storage::Default>& data__):
-		data_{&data__, 1u}
+		data_{&data__, 1u},
+		size_{data__.size()}
 	{}
 
 	auto& get_handle() {
@@ -25,6 +27,10 @@ public:
 		return data_;
 	}
 
+	uint64_t size() const {
+		return size_;
+	}
+
 	template<cl::sycl::access::mode AccessMode>
 	auto access(cl::sycl::handler& h){
 		return data_.template get_access<AccessMode>(h);