#include "sycl_basic.hpp"

saw::rpc_server<schema::BasicInterface, saw::encode::Native<saw::storage::Default>, saw::rmt::Sycl> listen_basic_sycl(saw::remote<saw::rmt::Sycl>& ctx, saw::remote_address<saw::rmt::Sycl>& addr){
	saw::interface<schema::BasicInterface, saw::encode::Native<saw::storage::Default>, cl::sycl::queue*> iface{
		[](saw::data<saw::schema::UInt64> in, cl::sycl::queue* q) -> saw::data<saw::schema::UInt64> {
			uint64_t inr = in.get();
			cl::sycl::buffer<uint64_t,1>  d_inc{ &inr, 1u };
			q->submit([&](cl::sycl::handler& h){
				auto a_inc = d_inc.get_access<cl::sycl::access::mode::read_write>(h);

				h.parallel_for(cl::sycl::range<1>(1u), [=] (cl::sycl::id<1> it){
					a_inc[0] += 1u;
				});
			});
			q->wait();
			return {inr};
		}
	};
	auto rpc_server = ctx.template listen<schema::BasicInterface, saw::encode::Native<saw::storage::Default>>(addr, std::move(iface));
	
	return rpc_server;
}