Fixed calc of sycl vals

1 files changed, 40 insertions, 7 deletions

diff --git a/modules/remote-sycl/c++/remote.hpp b/modules/remote-sycl/c++/remote.hpp
index bcc8a3c..4510237 100644
--- a/modules/remote-sycl/c++/remote.hpp
+++ b/modules/remote-sycl/c++/remote.hpp
@@ -22,22 +22,34 @@ class remote_data<T, Encoding, Storage, rmt::Sycl> {
 private:
 	id<T> id_;
 	id_map<T,Encoding,rmt::Sycl>* map_;
-	cl::sycl::queue* queue_;
 public:
 	/**
 	 * Main constructor
 	 */
 	remote_data(const id<T>& id, id_map<T, Encoding, rmt::Sycl>& map, cl::sycl::queue& queue__):
 		id_{id},
-		map_{&map},
-		queue_{&queue__}
+		map_{&map}
 	{}
 
 	/**
 	 * Wait for the data
 	 */
 	error_or<data<T,Encoding,Storage>> wait(){
+		auto eov = map_->find(id_);
+		if(eov.is_error()){
+			auto& err = eov.get_error();
+			return std::move(err);
+		}
+		auto& val = eov.get_value();
+		std::cout<<"Values Sycl in Map: "<<val->size()<<std::endl;
 
+		{
+			auto eocop = val->template copy_to_host<storage::Default>();
+			if(eocop.is_error()){
+				return eocop;
+			}
+			return eocop.get_value();
+		}
 	}
 
 	/**
@@ -87,7 +99,9 @@ public:
 	{
 		if(!device_data_){
 			total_length_ = 0u;
+			return;
 		}
+		queue_->wait();
 	}
 
 	template<typename Encoding, typename Storage>
@@ -102,6 +116,7 @@ public:
 			return;
 		}
 		queue_->template copy<data<T,encode::Native,storage::Default>>(&from.at(0), device_data_, total_length_);
+		queue_->wait();
 	}
 	
 	data(const data<Schema, encode::Native, rmt::Sycl>& from):
@@ -116,7 +131,10 @@ public:
 		// device_data_ = cl::sycl::malloc_device<typename native_data_type<T>::type>(from.size(), *queue_);
 		if(!device_data_){
 			total_length_ = 0u;
+			return;
 		}
+
+		queue_->template copy<data<T,encode::Native,storage::Default>>(from.device_data_, device_data_, total_length_);
 	}
 	
 	data<Schema, encode::Native, rmt::Sycl>& operator=(const data<Schema, encode::Native, rmt::Sycl>& rhs) {
@@ -158,7 +176,20 @@ public:
 		}
 	}
 
-	data<T, encode::Native, saw::storage::Default>& at(uint64_t i){
+	/**
+	 * Allocate appropriate meta data and then copy to host
+	 */
+	template<typename Storage>
+	error_or<data<Schema, encode::Native, Storage>> copy_to_host() const {
+		data<Schema,encode::Native, Storage> data_{total_length_};
+
+		/// TODO Check success
+		queue_->template copy<data<T,encode::Native,storage::Default>>(device_data_, &data_.at(0), total_length_);
+		queue_->wait();
+		return data_;
+	}
+
+	data<T, encode::Native, storage::Default>& at(uint64_t i){
 	//typename native_data_type<T>::type& at(uint64_t i){
 		return device_data_[i];
 	}
@@ -247,6 +278,9 @@ public:
 		return std::get<id_map<T,Encoding,rmt::Sycl>>(storage_.maps).next_free_id();
 	}
 
+	/**
+	 * Main constructor
+	 */
 	rpc_server(interface<Iface, Encoding, rmt::Sycl, InterfaceCtxT> cl_iface):
 		cmd_queue_{},
 		cl_interface_{std::move(cl_iface)},
@@ -255,12 +289,11 @@ public:
 
 	template<typename IdT, typename Storage>
 	remote_data<IdT, Encoding, Storage, rmt::Sycl> request_data(id<IdT> dat_id){
-		/// @TODO Fix so I can receive data
-		return {dat_id, std::get<id_map<IdT, Encoding,rmt::Sycl>>(storage_.maps)};
+		return {dat_id, std::get<id_map<IdT,Encoding,rmt::Sycl>>(storage_.maps), cmd_queue_};
 	}
 
 	/**
-	 * Rpc call
+	 * Rpc call based on the name
 	 */
 	template<string_literal Name, typename ClientAllocation>
 	error_or<