blob: e1a1e90bafed643eddc51bbc6aa80bd43c22ab4a (
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
|
#include "../c++/data.hpp"
namespace sch {
using namespace saw::schema;
}
template<typename T>
void inner_work(){
acpp::sycl::queue sycl_q;
constexpr uint64_t dat_size = 10000u;
data<sch::Array<T>, encode::Sycl<encode::Native>> dat{{{dat_size}},sycl_q};
data<sch::Ref<sch::Array<T>>, encode::Sycl<encode::Native>> dat_ref{dat};
auto dat_ptr = dat_ref.get_internal_data();
sycl_q.parallel_for(dat_size, [=](acpp::sycl::id<1> idx){
size_t i = idx[0];
dat_ptr[i] = {i};
}).wait();
for(uint64_t i = 0u; i < dat_size; ++i){
SAW_EXPECT(dat_ptr[i].get() == i, std::string{"Unexpected value: "} + std::to_string(i));
}
}
int main(){
using namespace saw;
inner_work<sch::Float32>();
inner_work<sch::Float64>();
inner_work<sch::MixedPrecision<sch::Float64,sch::Float32>>();
return 0;
}
|
