diff options
Diffstat (limited to 'modules/remote-sycl/c++/data.hpp')
| -rw-r--r-- | modules/remote-sycl/c++/data.hpp | 8 |
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); |