summaryrefslogtreecommitdiff
path: root/modules/remote-sycl/c++/data.hpp
blob: 91f74f81f445c4d2e557f717654899a287a296a9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
#pragma once

#include "common.hpp"

namespace saw {

/**
 * Generic wrapper class which stores data on the sycl side.
 * Most of the times this will be a root object.
 */
template<typename Schema>
class data<Schema, encode::Sycl<encode::Native>> {
private:
	cl::sycl::buffer<data<Schema, encode::Native>> data_;
	data<schema::UInt64, encode::Native> size_;
public:
	data(const data<Schema, encode::Native>& data__):
		data_{&data__, 1u},
		size_{data__.size()}
	{}

	auto& get_handle() {
		return data_;
	}

	const auto& get_handle() const {
		return data_;
	}

	data<schema::UInt64, encode::Native> size() const {
		return size_;
	}

	template<cl::sycl::access::mode AccessMode>
	auto access(cl::sycl::handler& h){
		return data_.template get_access<AccessMode>(h);
	}
	
	template<cl::sycl::access::mode AccessMode>
	auto access(cl::sycl::handler& h) const {
		return data_.template get_access<AccessMode>(h);
	}
};

template<typename Sch, uint64_t Dim>
class data<schema::Array<Sch, Dim>, encode::Sycl<encode::Native>> {
public:
	using Schema = schema::Array<Sch,Dim>;
private:
	cl::sycl::buffer<data<Sch, encode::Native>> data_;
	data<schema::UInt64, encode::Native> size_;
public:
	data(const data<Schema, encode::Native>& host_data__):
		data_{&host_data__.at({0u}),host_data__.size().get()},
		size_{host_data__.size()}
	{}

	auto& get_handle() {
		return data_;
	}

	const auto& get_handle() const {
		return data_;
	}

	data<schema::UInt64, encode::Native> size() const {
		return size_;
	}

	template<cl::sycl::access::mode AccessMode>
	auto access(cl::sycl::handler& h){
		return data_.template get_access<AccessMode>(h);
	}

	template<cl::sycl::access::mode AccessMode>
	auto access(cl::sycl::handler& h) const {
		return data_.template get_access<AccessMode>(h);
	}
};
}