GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-12 03:39:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Code cleanup

Browse Source
This commit is contained in:
Benoit Steiner 2016-11-18 16:58:09 -08:00
parent 1c6eafb46b
commit a357fe1fb9
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
unsupported/test/cxx11_tensor_device_sycl.cpp
View File

@ -31,10 +31,10 @@ void test_device_memory(const Eigen::SyclDevice &sycl_device) {
array<int, 1> tensorRange = {{sizeDim1}}; array<int, 1> tensorRange = {{sizeDim1}};
Tensor<DataType, 1, DataLayout> in(tensorRange); Tensor<DataType, 1, DataLayout> in(tensorRange);
Tensor<DataType, 1, DataLayout> in1(tensorRange); Tensor<DataType, 1, DataLayout> in1(tensorRange);
memset(in1.data(), 1,in1.size()*sizeof(DataType)); memset(in1.data(), 1, in1.size() * sizeof(DataType));
DataType * gpu_in_data = static_cast<DataType*>(sycl_device.allocate(in.size()*sizeof(DataType))); DataType* gpu_in_data = static_cast<DataType*>(sycl_device.allocate(in.size()*sizeof(DataType)));
sycl_device.memset(gpu_in_data, 1,in.size()*sizeof(DataType) ); sycl_device.memset(gpu_in_data, 1, in.size()*sizeof(DataType));
sycl_device.memcpyDeviceToHost(in.data(), gpu_in_data, in.size()*sizeof(DataType) ); sycl_device.memcpyDeviceToHost(in.data(), gpu_in_data, in.size()*sizeof(DataType));
for (int i=0; i<in.size(); i++) { for (int i=0; i<in.size(); i++) {
VERIFY_IS_EQUAL(in(i), in1(i)); VERIFY_IS_EQUAL(in(i), in1(i));
} }
@ -47,10 +47,13 @@ void test_device_exceptions(const Eigen::SyclDevice &sycl_device) {
int sizeDim1 = 100; int sizeDim1 = 100;
array<int, 1> tensorDims = {{sizeDim1}}; array<int, 1> tensorDims = {{sizeDim1}};
DataType* gpu_data = static_cast<DataType*>(sycl_device.allocate(sizeDim1*sizeof(DataType))); DataType* gpu_data = static_cast<DataType*>(sycl_device.allocate(sizeDim1*sizeof(DataType)));
TensorMap<Tensor<DataType, 1,DataLayout>> in(gpu_data, tensorDims); sycl_device.memset(gpu_data, 1, sizeDim1*sizeof(DataType));
TensorMap<Tensor<DataType, 1,DataLayout>> out(gpu_data, tensorDims);
TensorMap<Tensor<DataType, 1, DataLayout>> in(gpu_data, tensorDims);
TensorMap<Tensor<DataType, 1, DataLayout>> out(gpu_data, tensorDims);
out.device(sycl_device) = in / in.constant(0); out.device(sycl_device) = in / in.constant(0);
sycl_device.synchronize();
VERIFY(!sycl_device.ok()); VERIFY(!sycl_device.ok());
sycl_device.deallocate(gpu_data); sycl_device.deallocate(gpu_data);
} }